diff --git a/macrodensity/potential.py b/macrodensity/potential.py
index 5ee49f6..a964ae1 100644
--- a/macrodensity/potential.py
+++ b/macrodensity/potential.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 from macrodensity.io import read_vasp_density, read_gulp_potential  
-from ase import cube
+import ase
 
 
 def macroscopic_average(