diff --git a/setup.py b/setup.py
index d8d838e..67d6d93 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='xbTools',
-      version='0.0.5',
+      version='0.0.6',
       description='Toolbox to analyse and setup xbeach models',
       url='https://github.com/openearth/xbeach-toolbox',
       # Authors: Menno de Ridder (menno.deridder@deltares.nl), Cas van Bemmelen (cas.van.bemmelen@witteveenbos.com)
diff --git a/xbTools/xbeachtools.py b/xbTools/xbeachtools.py
index 6450425..84e5e93 100644
--- a/xbTools/xbeachtools.py
+++ b/xbTools/xbeachtools.py
@@ -15,7 +15,7 @@
 import matplotlib.pyplot as plt
 
 # import rotate grid from geometry
-from general.geometry import rotate_grid
+from .general.geometry import rotate_grid
     
 class XBeachModelSetup():
     '''