problem with the tests
fixing the tests