Fix previously failing test