diff --git a/superset-frontend/src/SqlLab/components/SaveQuery/index.tsx b/superset-frontend/src/SqlLab/components/SaveQuery/index.tsx
index f05c00818..684c4ed39 100644
--- a/superset-frontend/src/SqlLab/components/SaveQuery/index.tsx
+++ b/superset-frontend/src/SqlLab/components/SaveQuery/index.tsx
@@ -87,6 +87,7 @@ const SaveQuery = ({
   const query = useMemo(
     () => ({
       ...queryEditor,
+      columns,
     }),
     [queryEditor, columns],
   );