Removed accidental .only in tests