diff --git a/fltk/util/config/learning_config.py b/fltk/util/config/learning_config.py
index 2a53cc2f..ef508214 100644
--- a/fltk/util/config/learning_config.py
+++ b/fltk/util/config/learning_config.py
@@ -14,12 +14,8 @@
 # noinspection PyProtectedMember
 from torch.nn.modules.loss import _Loss
 
-from fltk.util.config.definitions import DataSampler, Loss, get_loss_function
-from fltk.util.config.definitions.aggregate import Aggregations
-from fltk.util.config.definitions.dataset import Dataset
-from fltk.util.config.definitions.logging import LogLevel
-from fltk.util.config.definitions.net import Nets
-from fltk.util.config.definitions.optim import Optimizations
+from fltk.util.config.definitions import (DataSampler, Loss, get_loss_function, Aggregations, Dataset, LogLevel, Nets,
+                                          Optimizations)
 
 
 def _eval_decoder(obj: Union[str, T]) -> Union[Any, T]: