fixing test failure when running all test