diff --git a/examples/classifier_compression/compress_classifier.py b/examples/classifier_compression/compress_classifier.py
index cc7965e717a06514530f45ae2859ab5d09ccd040..883ff9285a9ca68b01438155a1d1cbac12811fa6 100755
--- a/examples/classifier_compression/compress_classifier.py
+++ b/examples/classifier_compression/compress_classifier.py
@@ -196,8 +196,8 @@ def main():
 
     # This sample application can be invoked to produce various summary reports.
     if args.summary:
-        for s in which_summary:
-            distiller.model_summary(model, s, dataset)
+        for summary in args.summary:
+            distiller.model_summary(model, summary, args.dataset)
         return
 
     if args.export_onnx is not None: