Hackish fix for the tests.