[TEST] class tests fixed