diff --git a/ENDA.py b/ENDA.py
index 04b1f7dac911b9c6fce0bda64a34b4eadd0caf70..e093fc4af93d6e43a67d38adc4287de901e5f74e 100644
--- a/ENDA.py
+++ b/ENDA.py
@@ -8,7 +8,7 @@ from scipy.interpolate import CubicSpline
 from copy import deepcopy
 
 # Own modules
-from models import CBL
+from PE_CBL_models import CBL
 from observations import *
 
 # TODO: comprehensive documentation is missing.
diff --git a/verification.py b/verification.py
index e1e9646f32aaeafabadd181f96e0ab0ce01f0f1c..5a8f2f385782c26c0c792bce58ab82612c3c3d25 100644
--- a/verification.py
+++ b/verification.py
@@ -3,7 +3,7 @@
 import numpy as np
 from matplotlib import pyplot as p
 
-from models import CBL
+from PE_CBL_models import CBL
 from observations import decode_observations
 
 # Default CBL model settings