[TEST] class tests updated