From 7f9526bb81ab46785833d4f83cfd7d08301459af Mon Sep 17 00:00:00 2001
From: Adam Chalmers <adam.chalmers@zoo.dev>
Date: Fri, 24 Jan 2025 12:43:09 -0600
Subject: [PATCH] Fix revolve

---
 src/lang/modifyAst/addRevolve.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/lang/modifyAst/addRevolve.ts b/src/lang/modifyAst/addRevolve.ts
index 2ac2a90c3..dd6bc2017 100644
--- a/src/lang/modifyAst/addRevolve.ts
+++ b/src/lang/modifyAst/addRevolve.ts
@@ -7,6 +7,7 @@ import {
   CallExpression,
   PipeExpression,
   VariableDeclarator,
+  CallExpressionKw,
 } from 'lang/wasm'
 import { Selections } from 'lib/selections'
 import { Node } from 'wasm-lib/kcl/bindings/Node'
@@ -50,10 +51,10 @@ export function revolveSketch(
       clonedAst,
       edge.graphSelections[0]?.codeRef.range
     )
-    const lineNode = getNodeFromPath<CallExpression>(
+    const lineNode = getNodeFromPath<CallExpression | CallExpressionKw>(
       clonedAst,
       pathToAxisSelection,
-      'CallExpression'
+      ['CallExpression', 'CallExpressionKw']
     )
     if (err(lineNode)) return lineNode