diff --git a/Gemfile.lock b/Gemfile.lock
index 009c59b..54e2e9c 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    shale-builder (0.2.3)
+    shale-builder (0.2.4)
       booleans (>= 0.1)
       shale (< 2.0)
       sorbet-runtime (> 0.5)
diff --git a/lib/shale/builder/version.rb b/lib/shale/builder/version.rb
index ac4b831..9063fdf 100644
--- a/lib/shale/builder/version.rb
+++ b/lib/shale/builder/version.rb
@@ -2,6 +2,6 @@
 
 module Shale
   module Builder
-    VERSION = '0.2.3'
+    VERSION = '0.2.4'
   end
 end
diff --git a/lib/tapioca/dsl/compilers/shale.rb b/lib/tapioca/dsl/compilers/shale.rb
index 17f6b26..292cb35 100644
--- a/lib/tapioca/dsl/compilers/shale.rb
+++ b/lib/tapioca/dsl/compilers/shale.rb
@@ -59,18 +59,18 @@ def decorate
 
             setter_type, nilable = shale_type_to_sorbet_setter_type(attribute)
             if attribute.collection?
-              setter_type = "T.nilable(T::Array[#{return_type}])"
+              setter_type_str = "T.nilable(T::Array[#{setter_type}])"
             elsif nilable
-              setter_type = "T.nilable(#{return_type})"
+              setter_type_str = "T.nilable(#{setter_type})"
             else
-              setter_type = return_type.to_s
+              setter_type_str = setter_type.to_s
             end
 
             # setter
             mod.create_method(
               "#{attribute.name}=",
-              parameters: [create_param('value', type: setter_type)],
-              return_type: setter_type,
+              parameters: [create_param('value', type: setter_type_str)],
+              return_type: setter_type_str,
               comments: comments,
             )
           end