diff --git a/.vscode/settings.json b/.vscode/settings.json
index 76d5026..bd6f69d 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,7 +1,9 @@
 {
     "rust-analyzer.cargo.features": [
-        "ml", 
-        "ml_gpu", 
-        "plot"
+        //"ml", 
+        //"ml_gpu", 
+        //"plot",
+        "ml_infer",
+        "wasm"
     ]
 }
\ No newline at end of file
diff --git a/Cargo.toml b/Cargo.toml
index 13d33c1..6b3e5e0 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -50,7 +50,7 @@ ml_train = ["ml_base", "rand", "rayon", "burn-autodiff", "burn/train", "burn/std
 ml_infer = ["ml_base", "burn", "burn-ndarray"]
 ml_gpu = ["ml_train", "burn-tch", "burn-wgpu"]
 
-wasm = ["rodio/wasm-bindgen", "wasm-bindgen", "wasm-bindgen-futures", "js-sys", "console_error_panic_hook", "wee_alloc", "gloo-timers"]
+wasm = ["rodio/wasm-bindgen", "wasm-bindgen", "wasm-bindgen-futures", "js-sys", "console_error_panic_hook", "wee_alloc", "gloo-timers", "burn/wasm-sync"]
 
 plot = ["plotters"]
 
diff --git a/src/wasm.rs b/src/wasm.rs
index 46abdc0..6f17bbf 100644
--- a/src/wasm.rs
+++ b/src/wasm.rs
@@ -2,8 +2,6 @@
 //!
 //! This module contains the WASM wrappers / bindings for the rest of the library.
 
-use std::panic;
-
 use js_sys::{Array, Object, Reflect};
 use wasm_bindgen::{convert::RefFromWasmAbi, prelude::*};