Fixed a problem with the rake task