diff --git a/embedding_converter/src/dataset.py b/embedding_converter/src/dataset.py
index cdc867f..05595eb 100644
--- a/embedding_converter/src/dataset.py
+++ b/embedding_converter/src/dataset.py
@@ -2,7 +2,7 @@ import glob
 
 from torch import Tensor
 from torch.utils.data import Dataset
-from torchvision import transforms, io
+from torchvision import io, transforms
 
 from .types import Batch
 
diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py
index 09b30b2..b0cd6db 100644
--- a/face_swapper/src/dataset.py
+++ b/face_swapper/src/dataset.py
@@ -3,7 +3,7 @@ import random
 
 from torch import Tensor
 from torch.utils.data import Dataset
-from torchvision import transforms, io
+from torchvision import io, transforms
 
 from .types import Batch