Fix module unloading