diff --git a/tests/integration/test_simple.py b/tests/integration/test_simple.py
index b8f5bd3..9bb3139 100644
--- a/tests/integration/test_simple.py
+++ b/tests/integration/test_simple.py
@@ -421,16 +421,16 @@ def testEntry() -> i32x4:
 def test_imported():
     code_py = """
 @imported
-def helper() -> i32:
+def helper(mul: i32) -> i32:
     pass
 
 @exported
 def testEntry() -> i32:
-    return helper()
+    return helper(2)
 """
 
-    def helper() -> int:
-        return 4238
+    def helper(mul: int) -> int:
+        return 4238 * mul
 
     result = Suite(code_py, 'test_imported').run_code(
         runtime='wasmer',
@@ -439,4 +439,4 @@ def testEntry() -> i32:
         }
     )
 
-    assert 4238 == result.returned_value
+    assert 8476 == result.returned_value