Fixes the tests.