diff --git a/src/batched_denoising/batched_denoising.py b/src/batched_denoising/batched_denoising.py
index 04a155a..5d2e727 100644
--- a/src/batched_denoising/batched_denoising.py
+++ b/src/batched_denoising/batched_denoising.py
@@ -1,5 +1,5 @@
 import torch
-from control_base import ControlNodeBase
+from .control_base import ControlNodeBase
 import comfy.model_management
 import comfy.samplers
 import math
@@ -776,4 +776,4 @@ class StreamScheduler(ControlNodeBase):
         selected_sigmas = torch.tensor(
             selected_sigmas, dtype=torch.float32, device=device
         )
-        return (selected_sigmas,)
\ No newline at end of file
+        return (selected_sigmas,)