Add missing hooks back