diff --git a/README.md b/README.md
index a857d90..755196e 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@ Qibochem key features
 
 ## Installation
 
-Using poetry 
+Using poetry
 
 ```
 git clone https://github.com/qiboteam/qibochem.git
diff --git a/src/qibochem/__init__.py b/src/qibochem/__init__.py
index c32b8d5..f6ed0ee 100644
--- a/src/qibochem/__init__.py
+++ b/src/qibochem/__init__.py
@@ -2,9 +2,4 @@
 
 __version__ = im.version(__package__)
 
-from qibochem import (
-    driver,
-    ansatz,
-    measurement,
-)
-
+from qibochem import ansatz, driver, measurement
diff --git a/src/qibochem/driver/__init__.py b/src/qibochem/driver/__init__.py
index e890d2d..b0871e6 100644
--- a/src/qibochem/driver/__init__.py
+++ b/src/qibochem/driver/__init__.py
@@ -1,2 +1,2 @@
-from qibochem.driver.molecule import *
 from qibochem.driver.hamiltonian import *
+from qibochem.driver.molecule import *