diff --git a/wattile/models/base_model.py b/wattile/models/base_model.py
index 2130c0e8..3a309b29 100644
--- a/wattile/models/base_model.py
+++ b/wattile/models/base_model.py
@@ -9,9 +9,9 @@
 from torch.utils.tensorboard import SummaryWriter
 
 from wattile.error import ConfigsError
+from wattile.model_registry import ModelRegistry
 from wattile.util import factors
 from wattile.visualization import timeseries_comparison
-from wattile.model_registry import ModelRegistry
 
 global file_prefix
 logger = logging.getLogger(str(os.getpid()))