Fixing problem with loading modules