Fixed an error in test