diff --git a/windows-snapshot-tool/CMakeLists.txt b/windows-snapshot-tool/CMakeLists.txt
index fb80428..ab4f41c 100644
--- a/windows-snapshot-tool/CMakeLists.txt
+++ b/windows-snapshot-tool/CMakeLists.txt
@@ -6,6 +6,7 @@ set(EXECUTABLE_OUTPUT_PATH ${CMAKE_BINARY_DIR})
 set(LIBRARY_OUTPUT_PATH ${CMAKE_BINARY_DIR})
 
 add_definitions(-DUNICODE -D_UNICODE)
+SET( CMAKE_EXE_LINKER_FLAGS  "-mwindows" )
 
 include_directories("${PROJECT_SOURCE_DIR}")
 
diff --git a/windows-snapshot-tool/src/main.cpp b/windows-snapshot-tool/src/main.cpp
index e195f8e..250306e 100644
--- a/windows-snapshot-tool/src/main.cpp
+++ b/windows-snapshot-tool/src/main.cpp
@@ -215,6 +215,8 @@ void minimize()
 void restore()
 {
     ShowWindow(Hwnd, SW_SHOW);
+    WCHAR path[12];
+    getRaspiStorage(path);
     CaptureImage(Hwnd, L"file.bmp");
 }