Fix rake task - Update method name