Add #register wrapper