Fix so tests can all run again (remove .only)