From 6e9efb73a3135bb9790fbe19a580c167969a016a Mon Sep 17 00:00:00 2001
From: Serge <5920850+seerge@users.noreply.github.com>
Date: Tue, 26 Mar 2024 22:36:25 +0100
Subject: [PATCH] UI Tweaks

---
 app/Display/ScreenControl.cs | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/app/Display/ScreenControl.cs b/app/Display/ScreenControl.cs
index 7e5c21b1..9a0d1958 100644
--- a/app/Display/ScreenControl.cs
+++ b/app/Display/ScreenControl.cs
@@ -1,7 +1,4 @@
-using GHelper.Helpers;
-using System.Management;
-
-namespace GHelper.Display
+namespace GHelper.Display
 {
     public class ScreenControl
     {
@@ -28,10 +25,18 @@ namespace GHelper.Display
         public void SetScreen(int frequency = -1, int overdrive = -1, int miniled = -1)
         {
             var laptopScreen = ScreenNative.FindLaptopScreen(true);
-            if (laptopScreen is null) return;
+            if (laptopScreen is null)
+            {
+                InitScreen();
+                return;
+            }
 
             var refreshRate = ScreenNative.GetRefreshRate(laptopScreen);
-            if (refreshRate < 0) return;
+            if (refreshRate < 0)
+            {
+                InitScreen();
+                return;
+            }
 
             if (frequency >= MAX_REFRESH)
             {
@@ -49,7 +54,6 @@ namespace GHelper.Display
                 if (!AppConfig.IsOLED() && overdrive != Program.acpi.DeviceGet(AsusACPI.ScreenOverdrive))
                 {
                     Program.acpi.DeviceSet(AsusACPI.ScreenOverdrive, overdrive, "ScreenOverdrive");
-
                 }
             }
 
@@ -89,7 +93,7 @@ namespace GHelper.Display
             }
 
             AppConfig.Set("miniled", miniled);
-            SetScreen(-1, -1, miniled);
+            SetScreen(miniled: miniled);
             return miniled;
         }