Fixed an error in the unit test