Fixing another broken test