diff --git a/application/remote_desk/screen_capture/windows/screen_capture_wgc.h b/application/remote_desk/screen_capture/windows/screen_capture_wgc.h
index 9dc58d3..99610e7 100644
--- a/application/remote_desk/screen_capture/windows/screen_capture_wgc.h
+++ b/application/remote_desk/screen_capture/windows/screen_capture_wgc.h
@@ -6,6 +6,9 @@
 #include <string>
 #include <thread>
 
+#include "wgc_session.h"
+#include "wgc_session_impl.h"
+
 typedef struct {
   int left;
   int top;
@@ -16,7 +19,7 @@ typedef struct {
 typedef std::function<void(unsigned char *, int, int, int)> cb_desktop_data;
 typedef std::function<void(int)> cb_desktop_error;
 
-class ScreenCaptureWgc {
+class ScreenCaptureWgc : public WgcSession::wgc_session_observer {
  public:
   ScreenCaptureWgc();
   ~ScreenCaptureWgc();