1 files changed, 5 insertions, 6 deletions
diff --git a/libhwc2.1/libdevice/ExynosDeviceModule.h b/libhwc2.1/libdevice/ExynosDeviceModule.h
index ba37e29..6c772b6 100644
--- a/libhwc2.1/libdevice/ExynosDeviceModule.h
+++ b/libhwc2.1/libdevice/ExynosDeviceModule.h
@@ -17,9 +17,7 @@
 #ifndef EXYNOS_DEVICE_MODULE_H
 #define EXYNOS_DEVICE_MODULE_H
 
-#include <gs101/displaycolor/displaycolor_gs101.h>
-
-#include "DisplayColorLoader.h"
+#include "DisplayColorModule.h"
 #include "ExynosDevice.h"
 
 using namespace displaycolor;
@@ -27,19 +25,20 @@ using namespace displaycolor;
 namespace gs101 {
 
 class ExynosDeviceModule : public ExynosDevice {
+    using GsInterfaceType = gs::ColorDrmBlobFactory::GsInterfaceType;
     public:
         ExynosDeviceModule();
         virtual ~ExynosDeviceModule();
 
-        IDisplayColorGS101* getDisplayColorInterface() { return mDisplayColorInterface; }
+        GsInterfaceType* getDisplayColorInterface() { return mDisplayColorInterface; }
         void setActiveDisplay(uint32_t index) { mActiveDisplay = index; }
         uint32_t getActiveDisplay() const { return mActiveDisplay; }
 
     private:
         int initDisplayColor(const std::vector<displaycolor::DisplayInfo>& display_info);
 
-        IDisplayColorGS101* mDisplayColorInterface;
-        DisplayColorLoader mDisplayColorLoader;
+        GsInterfaceType * mDisplayColorInterface;
+        gs::ColorDrmBlobFactory::DcLoaderType mDisplayColorLoader;
         uint32_t mActiveDisplay;
 };