1 files changed, 10 insertions, 9 deletions
diff --git a/cli/ops/mod.rs b/cli/ops/mod.rs
index 53f74ad1f..562aa8649 100644
--- a/cli/ops/mod.rs
+++ b/cli/ops/mod.rs
@@ -10,17 +10,18 @@ pub mod bench;
 pub mod testing;
 
 pub fn cli_exts(ps: ProcState) -> Vec<Extension> {
-  vec![init_proc_state(ps)]
+  vec![deno_cli::init_ops(ps)]
 }
 
-fn init_proc_state(ps: ProcState) -> Extension {
-  Extension::builder("deno_cli")
-    .ops(vec![op_npm_process_state::decl()])
-    .state(move |state| {
-      state.put(ps.clone());
-    })
-    .build()
-}
+deno_core::extension!(deno_cli,
+  ops = [op_npm_process_state],
+  config = {
+    ps: ProcState,
+  },
+  state = |state, ps| {
+    state.put(ps);
+  },
+);
 
 #[op]
 fn op_npm_process_state(state: &mut OpState) -> Result<String, AnyError> {