Module loading fix