Fix for tests not passing.