diff --git a/reinforcement_learning.py b/reinforcement_learning.py
index 357ea7d..cef5545 100644
--- a/reinforcement_learning.py
+++ b/reinforcement_learning.py
@@ -158,7 +158,7 @@
 import numpy as np
 import tensorflow as tf
 import gym
-import scipy.ndimage
+import scipy
 import sys
 import os
 import time