lib/lit/builder/object.rb in litl-0.1.0 vs lib/lit/builder/object.rb in litl-0.1.1

@@ -141,10 +141,11 @@
           elsif type.is_a?(AST::Type::AnonymousEnum)
             make_enum(type.variants)
           elsif type == AST::Type::Primitive::String ||
                 type == AST::Type::Primitive::Integer ||
                 type == AST::Type::Primitive::Float ||
-                type == AST::Type::Primitive::Boolean
+                type == AST::Type::Primitive::Boolean ||
+                type == AST::Type::Primitive::Unit
             Module.new
           elsif type.is_a?(AST::Type::Alias)
             target_module = @target_module
 
             Class.new do