diff --git a/brushnet_nodes.py b/brushnet_nodes.py
index b333695..39c9b57 100644
--- a/brushnet_nodes.py
+++ b/brushnet_nodes.py
@@ -168,7 +168,7 @@ class BrushNet:
             is_SDXL = True
             if not brushnet["SDXL"]:
                 raise Exception("Base model is SDXL, but BrushNet is SD15 type")    
-            raise Exception("SDXL is not implemented yet")
+            #raise Exception("SDXL is not implemented yet")
         else:
             print('Base model type: ', type(model.model.model_config))
             raise Exception("Unsupported model type: " + str(type(model.model.model_config)))
@@ -217,11 +217,12 @@ class BrushNet:
 
         # apply patches to code
 
-        if 'BrushNet' not in nodes.common_ksampler.__doc__:
+        if nodes.common_ksampler.__doc__ is None or 'BrushNet' not in nodes.common_ksampler.__doc__:
             nodes.original_common_ksampler = nodes.common_ksampler
             nodes.common_ksampler = modified_common_ksampler
 
-        if 'BrushNet' not in comfy.ldm.modules.diffusionmodules.openaimodel.forward_timestep_embed.__doc__:
+        if comfy.ldm.modules.diffusionmodules.openaimodel.forward_timestep_embed.__doc__ is None or \
+            'BrushNet' not in comfy.ldm.modules.diffusionmodules.openaimodel.forward_timestep_embed.__doc__:
             comfy.ldm.modules.diffusionmodules.openaimodel.original_forward_timestep_embed =  \
             comfy.ldm.modules.diffusionmodules.openaimodel.forward_timestep_embed
             comfy.ldm.modules.diffusionmodules.openaimodel.forward_timestep_embed = modified_forward_timestep_embed
diff --git a/example/test_image3.png b/example/test_image3.png
new file mode 100644
index 0000000..6474048
Binary files /dev/null and b/example/test_image3.png differ
diff --git a/example/test_image4.png b/example/test_image4.png
new file mode 100644
index 0000000..ef0c781
Binary files /dev/null and b/example/test_image4.png differ
diff --git a/example/test_mask3.png b/example/test_mask3.png
new file mode 100644
index 0000000..98478a4
Binary files /dev/null and b/example/test_mask3.png differ