diff --git a/gridlock/__init__.py b/gridlock/__init__.py
index abcadc2..45d1aa8 100644
--- a/gridlock/__init__.py
+++ b/gridlock/__init__.py
@@ -15,8 +15,8 @@ Dependencies:
 - mpl_toolkits.mplot3d  [Grid.visualize_isosurface()]
 - skimage               [Grid.visualize_isosurface()]
 """
-from .error import GridError
-from .grid import Grid
+from .error import GridError as GridError
+from .grid import Grid as Grid
 
 __author__ = 'Jan Petykiewicz'
 __version__ = '1.1'