Fixes update rake task