Remove .only from one of the tests