Fix for broken tests.