Make rake task work