Removed .only from tests