diff --git a/setup.py b/setup.py
index cc5e82a..12c38ae 100644
--- a/setup.py
+++ b/setup.py
@@ -26,5 +26,5 @@
     install_requires=REQUIREMENTS,
     packages=find_packages(),
     package_dir={'torchxrayvision': 'torchxrayvision'},
-    package_data={'torchxrayvision': ['data/*.zip','data/*.gz','data/*.tgz','baseline_models/chexpert/*.json']},
+    package_data={'torchxrayvision': ['data/*.zip','data/*.gz','data/*.tgz','baseline_models/*/*.json','baseline_models/*/*/*.json']},
 )
diff --git a/torchxrayvision/_version.py b/torchxrayvision/_version.py
index 667c8b5..9621eef 100644
--- a/torchxrayvision/_version.py
+++ b/torchxrayvision/_version.py
@@ -1 +1 @@
-__version__ = "0.0.33"
+__version__ = "0.0.34"