Fix warning from incorrect hook signature