Make reloadModule work better