fixed test case failure