diff --git a/src/canari/unittests/maltego/entities.py b/src/canari/unittests/maltego/entities.py
index <HASH>..<HASH> 100644
--- a/src/canari/unittests/maltego/entities.py
+++ b/src/canari/unittests/maltego/entities.py
@@ -1,6 +1,6 @@
 from datetime import date, datetime, timedelta
 from canari.maltego.message import Entity, Field, StringEntityField, IntegerEntityField, FloatEntityField, \
-    BooleanEntityField, EnumEntityField, LongEntityField, DateTimeEntityField, DateEntityField, timespan, \
+    BooleanEntityField, EnumEntityField, LongEntityField, DateTimeEntityField, DateEntityField, TimeSpan, \
     TimeSpanEntityField, RegexEntityField, ColorEntityField
 from unittest import TestCase
Old entities unittests need to update