fix to the tests