[TEST] Fixed the bug in Test class