diff --git a/modules/impact/config.py b/modules/impact/config.py
index 88d0e82..80db901 100644
--- a/modules/impact/config.py
+++ b/modules/impact/config.py
@@ -2,7 +2,7 @@ import configparser
 import os
 
 
-version_code = [5, 5, 3]
+version_code = [5, 5, 4]
 version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 dependency_version = 20
diff --git a/modules/impact/impact_sampling.py b/modules/impact/impact_sampling.py
index c2600b7..cd843a4 100644
--- a/modules/impact/impact_sampling.py
+++ b/modules/impact/impact_sampling.py
@@ -100,14 +100,8 @@ def ksampler(sampler_name, total_sigmas, extra_options={}, inpaint_options={}):
 
         sampler_function = sample_dpmpp_sde
 
-    elif sampler_name == "uni_pc":
-        return samplers.KSAMPLER(samplers.uni_pc.sample_unipc)
-    elif sampler_name == "uni_pc_bh2":
-        return samplers.KSAMPLER(samplers.uni_pc.sample_unipc_bh2)
-    elif sampler_name == 'ddim':
-        return samplers.ksampler("euler", inpaint_options={"random": True})
     else:
-        return samplers.ksampler(sampler_name, extra_options, inpaint_options)
+        return comfy.samplers.sampler_object(sampler_name)
 
     return samplers.KSAMPLER(sampler_function, extra_options, inpaint_options)