diff --git a/a3c_main.py b/a3c_main.py
index 531ea51..3d28589 100644
--- a/a3c_main.py
+++ b/a3c_main.py
@@ -1,6 +1,7 @@
 import vizdoom
 import argparse
 import os
+os.environ["OMP_NUM_THREADS"] = "1"
 import numpy as np
 import torch
 import torch.multiprocessing as mp
@@ -12,8 +13,6 @@
 
 import logging
 
-os.environ["OMP_NUM_THREADS"] = "1"
-
 parser = argparse.ArgumentParser(description='Gated-Attention for Grounding')
 
 # Environment arguments