diff --git a/.gitmodules b/.gitmodules
index 8f5366a..aadde2d 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,3 +1,6 @@
 [submodule "mantra"]
 	path = mantra
 	url = git@github.com:aidos-lab/mantra.git
+[submodule "TopoBenchmarkX"]
+	path = TopoBenchmarkX
+	url = https://github.com/pyt-team/TopoBenchmarkX
diff --git a/TopoBenchmarkX b/TopoBenchmarkX
new file mode 160000
index 0000000..04b085f
--- /dev/null
+++ b/TopoBenchmarkX
@@ -0,0 +1 @@
+Subproject commit 04b085f729813c857542b6ee4a98514283b04a86
diff --git a/pyproject.toml b/pyproject.toml
index 0641f40..273b060 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ classifiers = [
 ]
 
 [tool.poetry.dependencies]
-python = "3.10.13"
+python = "~3.10"
 gudhi = "3.9.0"
 lightning = "2.2.5"
 wandb = "0.17.0"