Modify unit test for important attribute