Fix unregister script