fix unit test mistake