fix for the tests