Fix broken test :S