Fixed test cases

Fixed errors in test cases