fixing the test issue