Fix one more broken test