fixed tests
Fixed problem