diff --git a/build/src/lib.rs b/build/src/lib.rs
index 18447b37..29534b4c 100644
--- a/build/src/lib.rs
+++ b/build/src/lib.rs
@@ -93,7 +93,7 @@ cfg_if! {
   }
 }
 
-fn setup_napi_feature() {
+pub fn setup_napi_feature() {
   let napi_version = String::from_utf8(
     Command::new("node")
       .args(&["-e", "console.log(process.versions.napi)"])
diff --git a/napi/build.rs b/napi/build.rs
index 1f866b6a..1c4d57bc 100644
--- a/napi/build.rs
+++ b/napi/build.rs
@@ -1,5 +1,3 @@
-extern crate napi_build;
-
 fn main() {
-  napi_build::setup();
+  napi_build::setup_napi_feature();
 }