diff --git a/go.mod b/go.mod
index 70ebea90..728af063 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.26.1
 require (
 	github.com/apache/arrow-go/v18 v18.5.2
 	github.com/bradleyjkemp/cupaloy/v2 v2.8.0
-	github.com/cloudquery/codegen v0.4.0
+	github.com/cloudquery/codegen v0.4.1
 	github.com/cloudquery/plugin-sdk/v4 v4.95.0
 	github.com/goccy/go-json v0.10.6
 	github.com/invopop/jsonschema v0.13.0
diff --git a/go.sum b/go.sum
index fc32cdf6..7f5ecb7b 100644
--- a/go.sum
+++ b/go.sum
@@ -18,8 +18,8 @@ github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UF
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
 github.com/cloudquery/cloudquery-api-go v1.14.8 h1:iwXOQoVINrDQ69gtqBjWPSIdg7KcyLMq//FJkLxzUV4=
 github.com/cloudquery/cloudquery-api-go v1.14.8/go.mod h1:d+I8E+z3vmvTvCNXZ5YNhxV9InY/i1siXa0vCYq+ABk=
-github.com/cloudquery/codegen v0.4.0 h1:5UWbcNd9uBVYerMeeWdYC+8IZO9l/TEMqLQUD0JYOhs=
-github.com/cloudquery/codegen v0.4.0/go.mod h1:DHNcqoez8nVHVPHiOnbcDR/tXdAtLJngaCu2AOFlz90=
+github.com/cloudquery/codegen v0.4.1 h1:c9D18N925tUvnDeGHIl3JWKj37TyII9daHufkf8hU+Y=
+github.com/cloudquery/codegen v0.4.1/go.mod h1:QWIOD6R1aCa+YM+th+9Qt9lZw+ztdJR9JDEMLWyazwM=
 github.com/cloudquery/jsonschema v0.0.0-20260327151118-8dfb902740f6 h1:5KvFfR2HlsFFSUpnjrfp0D3jv69sYV3VlxEfmBd4EL0=
 github.com/cloudquery/jsonschema v0.0.0-20260327151118-8dfb902740f6/go.mod h1:0SoZ/U7yJlNOR+fWsBSeTvTbGXB6DK01tzJ7m2Xfg34=
 github.com/cloudquery/plugin-sdk/v4 v4.95.0 h1:r7QnJchvRypdIm2l1G9CMB6CJWJIa1vlC/b/H5EbvPI=