Update CI rake task