Removed .only in tests