Fix broken test, finally.