fixed tests

Hi.
When I run the tests, they will fail. I fixed it.