small fix add more hooks