Fix remaining broken test