Would it be possible to check just for the class?  I'm worried that this test will break if we add additional attributes or newlines.