diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
index c13489b7..d7e1f487 100644
--- a/src/vanna/base/base.py
+++ b/src/vanna/base/base.py
@@ -1701,7 +1701,7 @@ def ask(
                 return None
             else:
                 return sql, None, None
-        return sql, df, None
+        return sql, df, fig
 
     def train(
         self,