diff --git a/demos/regression.py b/demos/regression.py
index 48d477c53ac3b488546235682b6f0b91e626316a..b6e9e097bbef7e0001618395489961ad9727f421 100644
--- a/demos/regression.py
+++ b/demos/regression.py
@@ -3,7 +3,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 import torch
 import supr
-from tspn.utils import drawnow, arrange_figs
+from supr.utils import drawnow
 from scipy.stats import norm
 
 #%% Dataset
diff --git a/supr/utils.py b/supr/utils.py
index 790caf19f7be993df846506cc31cd9922af8a769..3ee9c1fbec440bae9a7e28525026f392056182c6 100644
--- a/supr/utils.py
+++ b/supr/utils.py
@@ -9,6 +9,7 @@ from PyQt5 import QtWidgets
 
 # %%
 def drawnow():
+    plt.show(block=False)
     plt.gcf().canvas.draw()
     plt.gcf().canvas.flush_events()
 
@@ -67,4 +68,4 @@ def local_scramble_2d(dist: float, dim: tuple):
 def bsgen(v, v0):
     while v > v0:
         yield v
-        v = math.ceil(v / 2)
\ No newline at end of file
+        v = math.ceil(v / 2)