diff --git a/ImageManager.cpp b/ImageManager.cpp
index ca401e9..2eb00e8 100644
--- a/ImageManager.cpp
+++ b/ImageManager.cpp
@@ -75,13 +75,13 @@ void ImageManager::ProcessImage(uint32_t width, uint32_t height, uint8_t *imgPoi
        if (area > fault_area)
        {
            drawContours(viewfinder, contours, i, Scalar(ZERO, BITS8, ZERO), 2);
-           //parent->fault_mgr.SaveImage(frame);
-           //((FaultFinder*)(parent()))->fault_mgr.SaveImage(frame);
+           ((FaultFinder*)(parent()))->fault_mgr->SaveImage();
        }
     }
 
     // Send image to other servers
-    SendImage(viewfinder);}
+    SendImage(viewfinder);
+}
 
 void ImageManager::SendImage(Mat image)
 {
diff --git a/ImageManager.hpp b/ImageManager.hpp
index 4cd104c..8796601 100644
--- a/ImageManager.hpp
+++ b/ImageManager.hpp
@@ -25,4 +25,6 @@ private:
     
 };
 
+#include "FaultFinder.hpp"
+
 #endif
\ No newline at end of file