diff --git a/crategen/converters/tes_converter.py b/crategen/converters/tes_converter.py
index 6701627..9a2614d 100644
--- a/crategen/converters/tes_converter.py
+++ b/crategen/converters/tes_converter.py
@@ -1,7 +1,7 @@
 from pydantic import ValidationError
 
-from ..models.tes_models import TESData
-from ..models.wrroc_models import WRROCDataTES
+from ..models.tes_models import TESData, TESInput, TESOutput, TESExecutor, TESResources, TESState, TESTaskLog
+from ..models.wrroc_models import WRROCDataTES, WRROCInputs, WRROCOutputs
 from .abstract_converter import AbstractConverter