diff --git a/package.json b/package.json
index 61d74fd..a74a27b 100644
--- a/package.json
+++ b/package.json
@@ -61,7 +61,7 @@
     "bundle": "vsce package"
   },
   "devDependencies": {
-    "@napi-rs/cli": "^1.0.0",
+    "@napi-rs/cli": "^2.16.3",
     "@types/mocha": "^10.0.1",
     "@types/node": "16.x",
     "@types/vscode": "^1.81.0",
@@ -74,7 +74,6 @@
     "typescript": "^5.1.6"
   },
   "dependencies": {
-    "@napi-rs/cli": "^2.16.3",
     "@vscode/vsce": "^2.22.0"
   }
 }
diff --git a/src/rust/lib.rs b/src/rust/lib.rs
index 8a8f675..43aadf9 100644
--- a/src/rust/lib.rs
+++ b/src/rust/lib.rs
@@ -4,6 +4,7 @@ pub mod client;
 pub mod workspace;
 pub mod cursor;
 pub mod buffer;
+pub mod op_cache;
 
 #[derive(Debug)]
 struct JsCodempError(codemp::Error);
diff --git a/src/rust/op_cache.rs b/src/rust/op_cache.rs
index 1c0cc9a..0523975 100644
--- a/src/rust/op_cache.rs
+++ b/src/rust/op_cache.rs
@@ -1,4 +1,5 @@
 use std::collections::HashSet;
+use napi_derive::napi;
 
 pub type OpTuple = (String, u32, String, u32);