diff --git a/pds/imageextractor.py b/pds/imageextractor.py
index 42af893..c937874 100644
--- a/pds/imageextractor.py
+++ b/pds/imageextractor.py
@@ -17,10 +17,11 @@
 
 try:
 	import Image
+	import ImageMath
 except ImportError:
 	from PIL import Image
+	from PIL import ImageMath
 
-import ImageMath
 
 from core.common import open_pds
 from core.parser import Parser