Add hook for pool allocation pass