diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py
index ddfb0edd15c37c2674198f3e9b77d31ef1907613..499abc9497b8b5c71fac65e5d2db6fc876547041 100755
--- a/distiller/apputils/image_classifier.py
+++ b/distiller/apputils/image_classifier.py
@@ -34,7 +34,7 @@ import distiller.quantization as quantization
 import distiller.models as models
 from distiller.models import create_model
 import parser
-import partial
+from functools import partial
 import operator
 import argparse
 from distiller.utils import float_range_argparse_checker as float_range