diff --git a/pina/utils.py b/pina/utils.py
index ed18af0..429985a 100644
--- a/pina/utils.py
+++ b/pina/utils.py
@@ -1,6 +1,8 @@
 """Utils module"""
 from functools import reduce
 
+from .label_tensor import LabelTensor
+
 
 def number_parameters(model, aggregate=True, only_trainable=True): #TODO: check
     """