diff --git a/api/__init__.py b/api/__init__.py
index b547e26..5552941 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -168,4 +168,4 @@ from .s39_user import get_user_schema, get_user, get_all_users
 
 from .s40_schema import get_scheme_schema, get_scheme, get_all_schemes
 
-from .s41_pipe_risk_probability import get_pipe_risk_probability_now, get_pipe_risk_probability, get_network_pipe_risk_probability_now, get_pipes_risk_probability
+from .s41_pipe_risk_probability import get_pipe_risk_probability_now, get_pipe_risk_probability, get_network_pipe_risk_probability_now, get_pipes_risk_probability, get_pipe_risk_probability_geometries