fixing the tests
fixing the tests again