diff --git a/src/qibochem/ansatz/__init__.py b/src/qibochem/ansatz/__init__.py
index faf8576..582bc09 100644
--- a/src/qibochem/ansatz/__init__.py
+++ b/src/qibochem/ansatz/__init__.py
@@ -1,23 +1,17 @@
 from qibochem.ansatz.basis_rotation import (
-    unitary_rot_matrix,
-    swap_matrices,
-    givens_rotation_parameters,
-    givens_rotation_gate,
     br_circuit,
+    givens_rotation_gate,
+    givens_rotation_parameters,
+    swap_matrices,
+    unitary_rot_matrix,
 )
-from qibochem.ansatz.hardware_efficient import (
-    hea,
-)
-from qibochem.ansatz.hf_reference import (
-    bk_matrix_power2,
-    bk_matrix,
-    hf_circuit,
-)
+from qibochem.ansatz.hardware_efficient import hea
+from qibochem.ansatz.hf_reference import bk_matrix, bk_matrix_power2, hf_circuit
 from qibochem.ansatz.ucc import (
     expi_pauli,
-    ucc_circuit,
-    mp2_amplitude,
     generate_excitations,
+    mp2_amplitude,
     sort_excitations,
     ucc_ansatz,
+    ucc_circuit,
 )
diff --git a/src/qibochem/driver/__init__.py b/src/qibochem/driver/__init__.py
index b3aacb7..f02c813 100644
--- a/src/qibochem/driver/__init__.py
+++ b/src/qibochem/driver/__init__.py
@@ -1,9 +1,7 @@
-from qibochem.driver.molecule import (
-    Molecule,
-)
 from qibochem.driver.hamiltonian import (
     fermionic_hamiltonian,
-    qubit_hamiltonian,
     parse_pauli_string,
+    qubit_hamiltonian,
     symbolic_hamiltonian,
 )
+from qibochem.driver.molecule import Molecule