section_mangler: Change API for SectionMangler::with_return_type

PLC-lang · Jun 11, 2024 · 7ad554f · 7ad554f
1 parent b36dcbd
commit 7ad554f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/compiler/section_mangler/src/lib.rs b/compiler/section_mangler/src/lib.rs
@@ -84,9 +84,11 @@ impl SectionMangler {
         }
     }
 
-    pub fn with_return_type(self, return_type: Option<Type>) -> SectionMangler {
+    pub fn with_return_type(self, return_type: Type) -> SectionMangler {
         match self {
-            SectionMangler::Function(f) => SectionMangler::Function(FunctionMangler { return_type, ..f }),
+            SectionMangler::Function(f) => {
+                SectionMangler::Function(FunctionMangler { return_type: Some(return_type), ..f })
+            }
             SectionMangler::Variable(_) => unreachable!("global variables do not have a return type."),
         }
     }