diff --git a/codegen/tests/test_functions.rs b/codegen/tests/test_functions.rs
index b61ae340..cfc73275 100644
--- a/codegen/tests/test_functions.rs
+++ b/codegen/tests/test_functions.rs
@@ -1,4 +1,5 @@
 use rhai::module_resolvers::*;
+use rhai::plugin::*;
 use rhai::{Array, Engine, EvalAltResult, Module, RegisterFn, FLOAT};
 
 pub mod raw_fn {