fix for the test