Fixes one of the tests