diff --git a/tick/survival/__init__.py b/tick/survival/__init__.py
index 8989bb04d..68c11a039 100644
--- a/tick/survival/__init__.py
+++ b/tick/survival/__init__.py
@@ -1,6 +1,7 @@
 # License: BSD 3 clause
 
 import tick.base
+import tick.base_model.build.base_model
 
 from .cox_regression import CoxRegression