fixing the test!