Replace deprecated hook usage with another hook

Change-Id: Ic97fc5cfb33f88f5290b4664652b7c6262568d55