Fix for tests to pass