diff --git a/tasm-lib/src/traits/function.rs b/tasm-lib/src/traits/function.rs
index 3bf5248a..ad7a79b6 100644
--- a/tasm-lib/src/traits/function.rs
+++ b/tasm-lib/src/traits/function.rs
@@ -11,6 +11,7 @@ use crate::snippet_bencher::write_benchmarks;
 use crate::snippet_bencher::BenchmarkCase;
 use crate::snippet_bencher::NamedBenchmarkResult;
 use crate::test_helpers::test_rust_equivalence_given_complete_state;
+use crate::InitVmState;
 use crate::VmHasher;
 
 use super::basic_snippet::BasicSnippet;
@@ -55,6 +56,17 @@ pub struct FunctionInitialState {
     pub memory: HashMap<BFieldElement, BFieldElement>,
 }
 
+impl From<FunctionInitialState> for InitVmState {
+    fn from(value: FunctionInitialState) -> Self {
+        let nd = NonDeterminism::default().with_ram(value.memory);
+        Self {
+            stack: value.stack,
+            nondeterminism: nd,
+            ..Default::default()
+        }
+    }
+}
+
 pub struct ShadowedFunction<F: Function + 'static> {
     pub function: Rc<RefCell<F>>,
 }