diff --git a/gflownet/envs/crystals/crystal.py b/gflownet/envs/crystals/crystal.py
index 723f29d3..d50955e5 100644
--- a/gflownet/envs/crystals/crystal.py
+++ b/gflownet/envs/crystals/crystal.py
@@ -11,13 +11,14 @@
 import numpy as np
 import pandas as pd
 import torch
+from torchtyping import TensorType
+from tqdm import tqdm
+
 from gflownet.envs.crystals.composition import Composition
 from gflownet.envs.crystals.lattice_parameters import PARAMETER_NAMES, LatticeParameters
 from gflownet.envs.crystals.spacegroup import SpaceGroup
 from gflownet.envs.stack import Stack
 from gflownet.utils.crystals.constants import TRICLINIC
-from torchtyping import TensorType
-from tqdm import tqdm
 
 
 class Crystal(Stack):