Fixed error in test case