Fix broken test...