diff --git a/selenium/setup.py b/selenium/setup.py
index c21c40e..8aefdf0 100644
--- a/selenium/setup.py
+++ b/selenium/setup.py
@@ -18,7 +18,7 @@
     license="MIT",
     install_requires=[
         "selenium>=3",
-        "lambdatest-sdk-utils==1.*"
+        "lambdatest-sdk-utils"
     ],
     python_requires='>=3.6',
     classifiers=[
diff --git a/utils/lambdatest_sdk_utils/rest.py b/utils/lambdatest_sdk_utils/rest.py
index cb1da9b..1aacb85 100644
--- a/utils/lambdatest_sdk_utils/rest.py
+++ b/utils/lambdatest_sdk_utils/rest.py
@@ -4,11 +4,9 @@
 
 logger = get_logger()
 
-SMART_UI_API = get_smart_ui_server_address()
-
 def is_smartui_enabled():
     try:
-        response = requests.get(f'{SMART_UI_API}/healthcheck')
+        response = requests.get(f'{get_smart_ui_server_address()}/healthcheck')
         response.raise_for_status()
         return True
     except Exception as e:
@@ -17,7 +15,7 @@ def is_smartui_enabled():
 
 def fetch_dom_serializer():
     try:
-        response = requests.get(f'{SMART_UI_API}/domserializer')
+        response = requests.get(f'{get_smart_ui_server_address()}/domserializer')
         response.raise_for_status()
         return response.json()
     except requests.exceptions.HTTPError as e:
@@ -32,7 +30,7 @@ def fetch_dom_serializer():
 
 def post_snapshot(snapshot,pkg,**kwargs):
     try:
-        response = requests.post(f'{SMART_UI_API}/snapshot', json={
+        response = requests.post(f'{get_smart_ui_server_address()}/snapshot', json={
             'snapshot': {
                 'dom' : snapshot['dom'],
                 'name' : snapshot['name'],