fixed error in tests