Fix unload error