The Count wasn't punished for his 'political poetry', he was being punished for being an aristocrat. The reason they didn't stand him against the wall and shoot him at the book's beginning was that he had written a poem in 1913 that was considered inspirational by the revolutionaries.