Fix rake task error