diff --git a/src/pytorch_ie/utils/hydra.py b/src/pytorch_ie/utils/hydra.py
index 36ed38f0..71ef6860 100644
--- a/src/pytorch_ie/utils/hydra.py
+++ b/src/pytorch_ie/utils/hydra.py
@@ -1,7 +1,7 @@
 # taken from hydra/_internal/instantiate/_instantiate2.py
 from typing import Any, Callable, Optional, Type, Union
 
-from pytorch_ie.core import Document
+from pytorch_ie.core.document import Document
 
 
 class HydraException(Exception):