Fix hook registration

Change-Id: I05e5b82fc8be1d0451b76cea76e25b955e910b9b