diff --git a/go.mod b/go.mod
index 68ba4cf..c9c2a9c 100644
--- a/go.mod
+++ b/go.mod
@@ -4,7 +4,7 @@ go 1.19
 
 require (
 	github.com/antlr/antlr4/runtime/Go/antlr/v4 v4.0.0-20230512164433-5d1fd1a340c9
-	github.com/kwilteam/kuneiform-grammar-go v0.3.2-0.20230710155215-f2e02ad890d5
+	github.com/kwilteam/kuneiform-grammar-go v0.4.0
 	github.com/kwilteam/kwil-db v0.4.1-0.20230629144950-2f7a1637e2cf
 	github.com/spf13/cobra v1.7.0
 	github.com/stretchr/testify v1.8.1
diff --git a/go.sum b/go.sum
index d6dd04b..8382fdd 100644
--- a/go.sum
+++ b/go.sum
@@ -10,8 +10,8 @@ github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2
 github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/kwilteam/action-grammar-go v0.0.0-20230626155541-72265621f427 h1:MYJuuqYXqJN7m4CwPrYPdsOCy48mtMTH4u8zXRySKtg=
 github.com/kwilteam/action-grammar-go v0.0.0-20230626155541-72265621f427/go.mod h1:2KYz1ittMk7BbKRJSnSnRSysrgDUUtNB1Ha47ZMdc6M=
-github.com/kwilteam/kuneiform-grammar-go v0.3.2-0.20230710155215-f2e02ad890d5 h1:Q1oW1JIlokVsVQlvdrfdWj692NNrBpcMGqQ0uyGv5IA=
-github.com/kwilteam/kuneiform-grammar-go v0.3.2-0.20230710155215-f2e02ad890d5/go.mod h1:uki0OHLCXttIQqy/DsJ5X9WaNtRf3HB/OmxCCd+XdZU=
+github.com/kwilteam/kuneiform-grammar-go v0.4.0 h1:eI3tIZbaSgQv/UR4Bks90/QDYKulpqBjGrtuNw5lF3A=
+github.com/kwilteam/kuneiform-grammar-go v0.4.0/go.mod h1:uki0OHLCXttIQqy/DsJ5X9WaNtRf3HB/OmxCCd+XdZU=
 github.com/kwilteam/kwil-db v0.4.1-0.20230629144950-2f7a1637e2cf h1:WeeZhKwt24vi+vBFAKS58gYFxG9t9Cwhn9p6TWsvVA0=
 github.com/kwilteam/kwil-db v0.4.1-0.20230629144950-2f7a1637e2cf/go.mod h1:UrDKyYMtpDR5jfBXz9wUdxhXxS6cK6vUyBltQ+OPles=
 github.com/kwilteam/sql-grammar-go v0.0.2 h1:Tg8xo/Pzd72QsO19wwNvFrpw2muuBaERP7XTv21U4Iw=