Add an unregister API