diff --git a/pvnet/models/multimodal/multimodal.py b/pvnet/models/multimodal/multimodal.py
index 5af9ae2c..358cdddb 100644
--- a/pvnet/models/multimodal/multimodal.py
+++ b/pvnet/models/multimodal/multimodal.py
@@ -2,10 +2,10 @@
 
 from collections import OrderedDict
 from typing import Optional
-from omegaconf import DictConfig
 
 import torch
 from ocf_datapipes.batch import BatchKey, NWPBatchKey
+from omegaconf import DictConfig
 from torch import nn
 
 import pvnet