diff --git a/src/protocol/tests/utils.rs b/src/protocol/tests/utils.rs
index c39dedfc5eeac57e4908f14b69524ce22b87bbd3..3767cdc6efdb6d417b055eb1724470c94929a0d4 100644
--- a/src/protocol/tests/utils.rs
+++ b/src/protocol/tests/utils.rs
@@ -788,7 +788,7 @@ impl<'a> ExpressionTester<'a> {
         let mono_proc = get_procedure_monomorph(&self.ctx.heap, &self.ctx.types, self.definition_id);
         let mono_index = mono_proc.monomorph_index;
         let mono_data = &self.ctx.heap[self.definition_id].as_procedure().monomorphs[mono_index as usize];
-        let expr_info = &mono_data.expr_info[self.var_expr.type_index as usize];
+        let expr_info = &mono_data.expr_info[self.expr.type_index() as usize];
         let concrete_type = &self.ctx.types.get_monomorph(expr_info.type_id).concrete_type;
 
         // Serialize and check type
@@ -819,7 +819,7 @@ fn get_procedure_monomorph<'a>(heap: &Heap, types: &'a TypeTable, definition_id:
     };
 
     let mono_index = types.get_procedure_monomorph_type_id(&definition_id, &func_type).unwrap();
-    let mono_data = types.get_procedure_monomorph(mono_index);
+    let mono_data = types.get_monomorph(mono_index).variant.as_procedure();
 
     mono_data
 }