Fix broken test caused by previous commit