diff --git a/src/rhai/screen.rs b/src/rhai/screen.rs
index 7b8107a..d85d665 100644
--- a/src/rhai/screen.rs
+++ b/src/rhai/screen.rs
@@ -12,15 +12,11 @@ fn screen_error_to_rhai_error<T>(result: anyhow::Result<T>) -> Result<T, Box<Eva
 
 #[allow(dead_code)]
 pub fn register(engine: &mut Engine) {
-    let mut module = Module::new();
-
-    module.set_native_fn("screen_new", |name: &str, cmd: &str| {
+    engine.register_fn("screen_new", |name: &str, cmd: &str| {
         screen_error_to_rhai_error(new_screen(name, cmd))
     });
 
-    module.set_native_fn("screen_kill", |name: &str| {
+    engine.register_fn("screen_kill", |name: &str| {
         screen_error_to_rhai_error(kill_screen(name))
     });
-
-    engine.register_static_module("screen", module.into());
 }
\ No newline at end of file