Fix update on unload module