Register unit manager callback