diff --git a/french_law/python/main.py b/french_law/python/main.py
index 08f24d4a..51d464a4 100755
--- a/french_law/python/main.py
+++ b/french_law/python/main.py
@@ -3,7 +3,7 @@
 from datetime import date
 from src.allocations_familiales import PriseEnCharge_Code, Collectivite_Code
 from src.api import allocations_familiales, Enfant
-from src.catala import LogEvent, LogEventCode, reset_log, retrieve_log
+from catala.runtime import LogEvent, LogEventCode, reset_log, retrieve_log
 import timeit
 import argparse
 from typing import List, Any
diff --git a/french_law/python/src/api.py b/french_law/python/src/api.py
index 1a8ebbd2..ea8cd245 100644
--- a/french_law/python/src/api.py
+++ b/french_law/python/src/api.py
@@ -1,4 +1,4 @@
-from .catala import *
+from catala.runtime import *
 from .allocations_familiales import Collectivite, Collectivite_Code, InterfaceAllocationsFamilialesIn, PriseEnCharge, interface_allocations_familiales, PriseEnCharge_Code, EnfantEntree, InterfaceAllocationsFamilialesIn