diff --git a/bmc/test/image_mock.hpp b/bmc/test/image_mock.hpp
index 2a967ac..1d4f726 100644
--- a/bmc/test/image_mock.hpp
+++ b/bmc/test/image_mock.hpp
@@ -2,6 +2,8 @@
 
 #include "image_handler.hpp"
 
+#include <memory>
+
 #include <gmock/gmock.h>
 
 namespace ipmi_flash
@@ -19,4 +21,9 @@
     MOCK_METHOD0(getSize, int());
 };
 
+std::unique_ptr<ImageHandlerMock> CreateImageMock()
+{
+    return std::make_unique<ImageHandlerMock>();
+}
+
 } // namespace ipmi_flash