diff --git a/lime/lime_image.py b/lime/lime_image.py
index 72dce4e3..c80ee155 100644
--- a/lime/lime_image.py
+++ b/lime/lime_image.py
@@ -170,6 +170,8 @@ def explain_instance(self, image, classifier_fn, labels=(1,),
         else:
             fudged_image[:] = hide_color
 
+        top = labels
+
         data, labels = self.data_labels(image, fudged_image, segments,
                                         classifier_fn, num_samples,
                                         batch_size=batch_size)