diff --git a/frontend/src/metabase/writeback/components/ActionCreator/QueryActionEditor.tsx b/frontend/src/metabase/writeback/components/ActionCreator/QueryActionEditor.tsx
index 9dc3fabb19fe03f03755f30e542211bc95d3cbd6..af087ad71aa17d40e492af169a2b1e357182cba8 100644
--- a/frontend/src/metabase/writeback/components/ActionCreator/QueryActionEditor.tsx
+++ b/frontend/src/metabase/writeback/components/ActionCreator/QueryActionEditor.tsx
@@ -1,8 +1,8 @@
 import React from "react";
 
-import type { DatasetQuery } from "metabase-types/types/Card";
-
 import NativeQueryEditor from "metabase/query_builder/components/NativeQueryEditor";
+import type Query from "metabase-lib/lib/queries/Query";
+
 import type Question from "metabase-lib/lib/Question";
 import { EditorContainer } from "./ActionCreator.styled";
 
@@ -18,8 +18,8 @@ export function QueryActionEditor({
       <NativeQueryEditor
         query={question.query()}
         viewHeight="full"
-        setDatasetQuery={(newQuery: DatasetQuery) =>
-          setQuestion(question.setDatasetQuery(newQuery))
+        setDatasetQuery={(newQuery: Query) =>
+          setQuestion(question.setQuery(newQuery))
         }
         enableRun={false}
         hasEditingSidebar={false} // TODO: make sidebar components work in popovers