diff --git a/trustgraph-cli/scripts/tg-invoke-document-rag b/trustgraph-cli/scripts/tg-invoke-document-rag
index 6a5443dc..759d4200 100755
--- a/trustgraph-cli/scripts/tg-invoke-document-rag
+++ b/trustgraph-cli/scripts/tg-invoke-document-rag
@@ -44,7 +44,7 @@ def main():
     # )
 
     parser.add_argument(
-        '-q', '--query',
+        '-q', '--question',
         required=True,
         help=f'Question to answer',
     )
diff --git a/trustgraph-flow/trustgraph/gateway/document_rag.py b/trustgraph-flow/trustgraph/gateway/document_rag.py
index 9f1428b4..94d8f788 100644
--- a/trustgraph-flow/trustgraph/gateway/document_rag.py
+++ b/trustgraph-flow/trustgraph/gateway/document_rag.py
@@ -23,7 +23,7 @@ class DocumentRagRequestor(ServiceRequestor):
             query=body["query"],
             user=body.get("user", "trustgraph"),
             collection=body.get("collection", "default"),
-            doc_limit=body.get("doc-limit", 20),
+            doc_limit=int(body.get("doc-limit", 20)),
         )
 
     def from_response(self, message):