diff --git a/edgetpuvision/classify.py b/edgetpuvision/classify.py
index dd7bcc5..9f12af5 100644
--- a/edgetpuvision/classify.py
+++ b/edgetpuvision/classify.py
@@ -126,7 +126,7 @@
         inference_rate = next(fps_counter)
         if draw_overlay:
             start = time.monotonic()
-            results = engine.ClassifyWithInputTensor(tensor, threshold=args.threshold, top_k=args.top_k)
+            results = engine.classify_with_input_tensor(tensor, threshold=args.threshold, top_k=args.top_k)
             inference_time = time.monotonic() - start
 
             results = [(labels[i], score) for i, score in results]