diff --git a/__init__.py b/__init__.py
index 9baf0f9..99782cd 100644
--- a/__init__.py
+++ b/__init__.py
@@ -55,11 +55,10 @@ for model in model_config:
         add_static_resource(path, path, limit=True)
 
 # get comfyui revision
-from .py.libs.utils import get_comfyui_revision
+from .py.libs.utils import compare_revision
 
-comfyui_revision = get_comfyui_revision()
 new_frontend_revision = 2546
-web_default_version = 'v2' if int(comfyui_revision) >= new_frontend_revision else 'v1'
+web_default_version = 'v2' if compare_revision(new_frontend_revision) else 'v1'
 # web directory
 config_path = os.path.join(cwd_path, "config.yaml")
 if os.path.isfile(config_path):