diff --git a/build_tests.sh b/build_tests.sh
index b45973f..70cd252 100755
--- a/build_tests.sh
+++ b/build_tests.sh
@@ -1,6 +1,8 @@
 #!/bin/bash
 
-rustup target add wasm32-wasi
+rustup target add wasm32-unknown-unknown
+
+cargo install wasi2ic
 
 cd src/tests/fs_benchmark_test
 
diff --git a/src/tests/fs_benchmark_test/src/fs_benchmark_test_backend/src/lib.rs b/src/tests/fs_benchmark_test/src/fs_benchmark_test_backend/src/lib.rs
index 7ab60c8..9587fbc 100644
--- a/src/tests/fs_benchmark_test/src/fs_benchmark_test_backend/src/lib.rs
+++ b/src/tests/fs_benchmark_test/src/fs_benchmark_test_backend/src/lib.rs
@@ -1,6 +1,6 @@
 use std::{cell::RefCell, str::FromStr};
 
-use stable_fs::{fs::{DstBuf, FdStat, FileSystem, OpenFlags, SrcBuf, Whence}, storage::{stable::StableStorage, types::MAX_FILE_NAME}};
+use stable_fs::{fs::{DstBuf, FdStat, FileSystem, OpenFlags, SrcBuf, Whence}, storage::stable::StableStorage};
 use ic_stable_structures::{memory_manager::{MemoryId, MemoryManager}, DefaultMemoryImpl, Memory};
 
 
@@ -255,7 +255,7 @@ fn create_depth_folders(path: String, count: usize) -> String {
             dir_name = format!("{}/d{}", dir_name, num);
         }
 
-        fs.create_dir(root_dir, dir_name.as_str(), FdStat::default(), 0);
+        fs.create_dir(root_dir, dir_name.as_str(), FdStat::default(), 0).unwrap();
 
         format!("{}/{}", path, dir_name)