diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py
index 6012936..02d093a 100644
--- a/embedding_converter/src/training.py
+++ b/embedding_converter/src/training.py
@@ -121,7 +121,7 @@ def create_trainer() -> Trainer:
 
 
 def train() -> None:
-	config =\
+	config : Config =\
 	{
 		'dataset':
 		{