Fix the one remaining broken test