Fix tests (oops)