diff --git a/src/unity/python/turicreate/toolkits/image_classifier/_annotate.py b/src/unity/python/turicreate/toolkits/image_classifier/_annotate.py
index <HASH>..<HASH> 100644
--- a/src/unity/python/turicreate/toolkits/image_classifier/_annotate.py
+++ b/src/unity/python/turicreate/toolkits/image_classifier/_annotate.py
@@ -121,10 +121,13 @@ def annotate(data, image_column=None, annotation_column='annotations'):
 
     """
 
-    image_column = _tkutl._find_only_image_column(data) 
+    
 
     # Check Value of Column Variables
     if image_column == None:
+        image_column = _tkutl._find_only_image_column(data) 
+
+    if image_column == None:
         raise ValueError("'image_column' cannot be 'None'")
 
     if type(image_column) != str:
Image classification/annotation multiple image columns (#<I>)

* Added fix for the case that multiple image columns exist in the data