diff --git a/RGBController/RGBController_RGBFusion2.cpp b/RGBController/RGBController_RGBFusion2.cpp
index bd58a9207..3c7c80c3b 100644
--- a/RGBController/RGBController_RGBFusion2.cpp
+++ b/RGBController/RGBController_RGBFusion2.cpp
@@ -9,21 +9,6 @@
 
 #include "RGBController_RGBFusion2.h"
 
-int RGBController_RGBFusion2::GetMode()
-{
-    return(rgb_fusion->GetMode());
-}
-
-void RGBController_RGBFusion2::SetMode(int mode)
-{
-    rgb_fusion->SetMode(mode);
-}
-
-void RGBController_RGBFusion2::SetCustomMode()
-{
-    rgb_fusion->SetMode(0);
-}
-
 void RGBController_RGBFusion2::UpdateLEDs()
 {
     for (std::size_t led = 0; led < colors.size(); led++)
@@ -98,3 +83,13 @@ RGBController_RGBFusion2::RGBController_RGBFusion2(RGBFusion2Controller* rgb_fus
         zones.push_back(*new_zone);
     }
 }
+
+void RGBController_RGBFusion2::SetCustomMode()
+{
+    active_mode = 0;
+}
+
+void RGBController_RGBFusion2::UpdateMode()
+{
+    rgb_fusion->SetMode(modes[active_mode].value);
+}
\ No newline at end of file
diff --git a/RGBController/RGBController_RGBFusion2.h b/RGBController/RGBController_RGBFusion2.h
index c4bb3ca98..29497a6eb 100644
--- a/RGBController/RGBController_RGBFusion2.h
+++ b/RGBController/RGBController_RGBFusion2.h
@@ -16,13 +16,13 @@ class RGBController_RGBFusion2 : public RGBController
 {
 public:
     RGBController_RGBFusion2(RGBFusion2Controller* rgb_fusion_ptr);
-    int         GetMode();
-    void        SetMode(int mode);
-    void        SetCustomMode();
     void        UpdateLEDs();
     void        UpdateZoneLEDs(int zone);
     void        UpdateSingleLED(int led);
 
+    void        SetCustomMode();
+    void        UpdateMode();
+
 private:
     RGBFusion2Controller* rgb_fusion;
 };