diff --git a/darwin/dataset/utils.py b/darwin/dataset/utils.py
index 1b12cde4a..f0e2d8686 100644
--- a/darwin/dataset/utils.py
+++ b/darwin/dataset/utils.py
@@ -496,6 +496,8 @@ def compute_distributions(
     for partition in partitions:
         for annotation_type in annotation_types:
             split_file: Path = split_path / f"stratified_{annotation_type}_{partition}.txt"
+            if not split_file.exists():
+                split_file = split_path / f"random_{partition}.txt"
             stems: List[str] = [e.rstrip("\n\r") for e in split_file.open()]
 
             for stem in stems:
diff --git a/setup.py b/setup.py
index baf24ca03..5f58bc51f 100644
--- a/setup.py
+++ b/setup.py
@@ -47,6 +47,7 @@
         "darwin.exporter",
         "darwin.importer.formats",
         "darwin.exporter.formats",
+        "darwin.version",
     ],
     entry_points={"console_scripts": ["darwin=darwin.cli:main"]},
     classifiers=["Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License"],