Fix rake plugin