diff --git a/opencl_fdfd/__init__.py b/opencl_fdfd/__init__.py
index a6f9b28..bd4a6ff 100644
--- a/opencl_fdfd/__init__.py
+++ b/opencl_fdfd/__init__.py
@@ -37,7 +37,7 @@
     - jinja2
 """
 
-from .main import cg_solver
+from .main import cg_solver as cg_solver
 
 __author__ = 'Jan Petykiewicz'
 __version__ = '0.4'