diff --git a/seminar/treelib.py b/seminar/treelib.py
index 6efab260..8784c2d9 100644
--- a/seminar/treelib.py
+++ b/seminar/treelib.py
@@ -1,5 +1,6 @@
 from django.core.exceptions import ObjectDoesNotExist
 from django.db import transaction
+from seminar.models import TreeNode
 # NOTE: node.prev a node.succ jsou implementovány přímo v models.TreeNode
 # TODO: Všechny tyto funkce se naivně spoléhají na to, že jako parametr dostanou nějaký TreeNode (některé možná zvládnou i None)
 # TODO: Chceme, aby všechno nějak zvládlo None jako parametr.