diff --git a/GANDLF/utils/pydantic_config.py b/GANDLF/utils/pydantic_config.py
index 6e0bf6e4a..f8ef4bf9f 100644
--- a/GANDLF/utils/pydantic_config.py
+++ b/GANDLF/utils/pydantic_config.py
@@ -1,5 +1,5 @@
-from pydantic import BaseModel, ConfigDict
-from typing import Dict, Any, List, Optional, Field
+from pydantic import BaseModel, ConfigDict, Field
+from typing import Dict, Any, List, Optional
 from enum import Enum
 from GANDLF.models.modelBase import ModelBase
 from typing import Union