Little update for module restriction mode