Update unit tests to include attribute class