Fix test that's breaking build