Can't use return in rake task, use conditional instead