diff --git a/test/unit-tests.html b/test/unit-tests.html
index 64553121..6829bff7 100644
--- a/test/unit-tests.html
+++ b/test/unit-tests.html
@@ -8,27 +8,6 @@
     <script>
       mocha.setup('bdd');
     </script>
-
-    <script>
-      afterEach(function () {
-        var isFailed = this.currentTest.state == 'failed';
-        var isManual = this.currentTest.parent.title === 'Manual';
-
-        Konva.stages.forEach(function (stage) {
-          clearTimeout(stage.dblTimeout);
-        });
-
-        if (!isFailed && !isManual) {
-          Konva.stages.forEach(function (stage) {
-            stage.destroy();
-          });
-          if (Konva.DD._dragElements.size) {
-            throw 'Why drag elements are not cleaned?';
-          }
-        }
-      });
-    </script>
-
     <script>
       // CORE
       import './unit/Animation-test.ts';
diff --git a/test/unit/test-utils.ts b/test/unit/test-utils.ts
index cb6b130b..4f5aaf07 100644
--- a/test/unit/test-utils.ts
+++ b/test/unit/test-utils.ts
@@ -18,6 +18,25 @@ beforeEach(function () {
   Konva['inDblClickWindow'] = false;
 });
 
+// clear after test
+afterEach(function () {
+  var isFailed = this.currentTest.state == 'failed';
+  var isManual = this.currentTest.parent.title === 'Manual';
+
+  Konva.stages.forEach(function (stage) {
+    clearTimeout(stage.dblTimeout);
+  });
+
+  if (!isFailed && !isManual) {
+    Konva.stages.forEach(function (stage) {
+      stage.destroy();
+    });
+    if (Konva.DD._dragElements.size) {
+      throw 'Why drag elements are not cleaned?';
+    }
+  }
+});
+
 export const isNode = typeof global.document === 'undefined';
 export const isBrowser = !isNode;