Updated hook to include additional argument.

Signed-off-by: Zane M. Kolnik <zanematthew@gmail.com>