diff --git a/src/vanna/__init__.py b/src/vanna/__init__.py
index 34b15e5c..fde17752 100644
--- a/src/vanna/__init__.py
+++ b/src/vanna/__init__.py
@@ -815,7 +815,7 @@ def train(question: str = None, sql: str = None, ddl: str = None, documentation:
 
     if documentation:
         print("Adding documentation....")
-        return add_documentation(sql)
+        return add_documentation(documentation)
 
     if sql:
         if question is None: