diff --git a/rltorch/action_selector/EpsilonGreedySelector.py b/rltorch/action_selector/EpsilonGreedySelector.py
index 7bb04c1..ec92498 100644
--- a/rltorch/action_selector/EpsilonGreedySelector.py
+++ b/rltorch/action_selector/EpsilonGreedySelector.py
@@ -1,4 +1,5 @@
-from .ArgMaxSelector import ArgMaxSelector 
+from .ArgMaxSelector import ArgMaxSelector
+import numpy as np 
 class EpsilonGreedySelector(ArgMaxSelector):
     def __init__(self, model, action_size, device = None, epsilon = 0.1, epsilon_decay = 1, epsilon_min = 0.1):
         super(EpsilonGreedySelector, self).__init__(model, action_size, device = device)