Fix broken test: