Fix for broken test.