diff --git a/dynthres_unipc.py b/dynthres_unipc.py
index 8da9130..b90d9de 100644
--- a/dynthres_unipc.py
+++ b/dynthres_unipc.py
@@ -6,12 +6,7 @@ from modules import shared
 try:
     from modules.models.diffusion import uni_pc
 except Exception as e:
-    from modules.unipc import sampler
-    class VladUniPC(object):
-        def __init__(self):
-            self.sampler = sampler
-            self.uni_pc = sampler
-    uni_pc = VladUniPC()
+    from modules import unipc as uni_pc
 
 ######################### UniPC Implementation logic #########################