fixed error in unit test