Optimize module loading across forking and non-forking scenarios
