Update rake task
simple fix