diff --git a/gnes/client/cli.py b/gnes/client/cli.py
index 6a8abade..2ed32016 100644
--- a/gnes/client/cli.py
+++ b/gnes/client/cli.py
@@ -45,7 +45,7 @@ def __init__(self, args):
                 resp = list(stub.StreamCall(RequestGenerator.train(all_bytes, args.batch_size)))[-1]
                 print(resp)
             elif args.mode == 'index':
-                resp = list(stub.StreamCall(RequestGenerator.train(all_bytes, args.batch_size)))[-1]
+                resp = list(stub.StreamCall(RequestGenerator.index(all_bytes, args.batch_size)))[-1]
                 print(resp)
             elif args.mode == 'query':
                 for idx, q in enumerate(all_bytes):