diff --git a/main.py b/main.py
index 965a312..324252e 100644
--- a/main.py
+++ b/main.py
@@ -15,17 +15,16 @@
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument('query',
+    parser.add_argument('input',
                         type=str,
                         default='How much is the minimum guarantee payable by adidas?',
                         help='Enter the query to pass into the LLM')
     args = parser.parse_args()
-    query = args.query
-
+    
     # Setup DBQA
     start = timeit.default_timer()
     dbqa = setup_dbqa()
-    response = dbqa({'query': query})
+    response = dbqa({'query': args.input})
     end = timeit.default_timer()
 
     print(response['result'])