Register for callback on unplug event