diff --git a/bin/core/imag-link/src/lib.rs b/bin/core/imag-link/src/lib.rs
index bfa927dd..47851127 100644
--- a/bin/core/imag-link/src/lib.rs
+++ b/bin/core/imag-link/src/lib.rs
@@ -396,6 +396,7 @@ mod tests {
         modulename mock;
         version env!("CARGO_PKG_VERSION");
         with help "imag-link mocking app";
+        with ui builder function crate::ui::build_ui;
     }
     use self::mock::generate_test_runtime;
     use self::mock::reset_test_runtime;