fixed error in unit tests