Update rake task to pass CI