diff --git a/compiler/test_gen/src/helpers/wasm32_test_result.rs b/compiler/test_gen/src/helpers/wasm32_test_result.rs
index e76581eefa..34e2a86bea 100644
--- a/compiler/test_gen/src/helpers/wasm32_test_result.rs
+++ b/compiler/test_gen/src/helpers/wasm32_test_result.rs
@@ -14,8 +14,9 @@ pub trait Wasm32TestResult {
         wrapper_name: &str,
         main_function_index: u32,
     ) {
-        // Assumes the main function was the first one to be generated
-        let index = main_function_index + module.code.code_builders.len() as u32;
+        let index = module.import.function_count()
+            + module.code.preloaded_count
+            + module.code.code_builders.len() as u32;
 
         module.add_function_signature(Signature {
             param_types: Vec::with_capacity_in(0, arena),