1 files changed, 4 insertions, 3 deletions
diff --git a/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java b/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
index fc311b0..e259526 100644
--- a/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
+++ b/src/main/java/com/mavlushechka/a1qa/driverUtils/WebDriverFactory.java
@@ -1,5 +1,6 @@
 package com.mavlushechka.a1qa.driverUtils;
 
+import com.mavlushechka.a1qa.models.Browser;
 import com.mavlushechka.a1qa.utils.JSONParser;
 import com.mavlushechka.a1qa.utils.LoggerUtils;
 import com.mavlushechka.a1qa.utils.StringUtils;
@@ -16,12 +17,12 @@ public class WebDriverFactory {
     }
 
     protected static WebDriver createWebDriver() {
-        switch (StringUtils.convertToConstantCase(JSONParser.parseData("config", "browser.name"))) {
-            case "CHROME" -> {
+        switch (Browser.valueOf(StringUtils.convertToConstantCase(JSONParser.parseData("config", "browser.name")))) {
+            case CHROME -> {
                 LoggerUtils.info("Creating ChromeDriver.");
                 return createChromeDriver();
             }
-            case "FIREFOX" -> {
+            case FIREFOX -> {
                 LoggerUtils.info("Creating FirefoxDriver.");
                 return createFirefoxDriver();
             }