fixes the test that is failing