Fix import rake task