Fix unload error

Fixes the following: http://www.joduska.me/forum/topic/14803-unload-assembly-crash/