diff --git a/et_replay/pyproject.toml b/et_replay/pyproject.toml
index 206b03dd..d9c9f82e 100644
--- a/et_replay/pyproject.toml
+++ b/et_replay/pyproject.toml
@@ -7,8 +7,10 @@ name = "et_replay"
 version = "0.5.0"
 
 [tool.setuptools.package-dir]
+"et_replay.lib" = "lib"
 "et_replay.lib.comm" = "lib/comm"
 "et_replay.tools" = "tools"
+"param_bench" = "../../param"
 
 [project.scripts]
 comm_replay = "et_replay.tools.comm_replay:main"