[BUG] Remove bug from rake task