diff --git a/gb_code/gb_generator.py b/gb_code/gb_generator.py
index 74cc195..d9726b4 100644
--- a/gb_code/gb_generator.py
+++ b/gb_code/gb_generator.py
@@ -18,7 +18,7 @@
 import numpy as np
 from numpy import dot, cross
 from numpy.linalg import det, norm
-import csl_generator as cslgen
+from . import csl_generator as cslgen
 import warnings