fix for the test fix