Fix rake task for dev and prod