diff --git a/hippynn/molecular_dynamics/__init__.py b/hippynn/molecular_dynamics/__init__.py
index c7138a1f..d54a269b 100644
--- a/hippynn/molecular_dynamics/__init__.py
+++ b/hippynn/molecular_dynamics/__init__.py
@@ -1,6 +1,5 @@
-from .md import Variable, NullUpdater, VelocityVerlet, LangevinDynamics, MolecularDynamics
-
 """
 Molecular dynamics driver with great flexibility and customizability regarding which quantities which are evolved 
 and what algorithms are used to evolve them. Calls a hippynn `Predictor` on current state during each MD step. 
-"""
\ No newline at end of file
+"""
+from .md import Variable, NullUpdater, VelocityVerlet, LangevinDynamics, MolecularDynamics
\ No newline at end of file