Add a missing $args array to a couple hooks