Emergency fix. Didn't rerun tests and it broke.