diff --git a/merlin/models/torch/transforms/sequences.py b/merlin/models/torch/transforms/sequences.py
index 0706e3f916..b393e83396 100644
--- a/merlin/models/torch/transforms/sequences.py
+++ b/merlin/models/torch/transforms/sequences.py
@@ -20,8 +20,8 @@
 from torch import nn
 
 from merlin.models.torch.batch import Batch, Sequence
-from merlin.models.torch.schema import Selection, select
 from merlin.models.torch.block import BatchBlock
+from merlin.models.torch.schema import Selection, select
 from merlin.schema import Schema, Tags