diff --git a/bin/core/imag-store/src/create.rs b/bin/core/imag-store/src/create.rs
index 0384310f..c4a8ad14 100644
--- a/bin/core/imag-store/src/create.rs
+++ b/bin/core/imag-store/src/create.rs
@@ -168,6 +168,7 @@ mod tests {
         modulename mock;
         version env!("CARGO_PKG_VERSION");
         with help "imag-store mocking app";
+        with ui builder function crate::ui::build_ui;
     }
     use self::mock::generate_test_runtime;
 
diff --git a/bin/core/imag-store/src/delete.rs b/bin/core/imag-store/src/delete.rs
index b6189414..e361d535 100644
--- a/bin/core/imag-store/src/delete.rs
+++ b/bin/core/imag-store/src/delete.rs
@@ -49,6 +49,7 @@ mod tests {
         modulename mock;
         version env!("CARGO_PKG_VERSION");
         with help "imag-store mocking app";
+        with ui builder function crate::ui::build_ui;
     }
     use self::mock::generate_test_runtime;
     use self::mock::reset_test_runtime;