Fix the last broken test