Add args to some hooks