diff --git a/pyproject.toml b/pyproject.toml
index 0fb212e..1186474 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,6 +20,7 @@ dependencies = [
     'boto3',
     'pydantic>=2.0',
     'pydantic-settings>=2.0',
+    'aind-data-schema==0.33.3',
     'aind-data-transfer-models==0.3.0'
 ]
 
diff --git a/tests/test_models.py b/tests/test_models.py
index 92d77a0..313cb6b 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -42,7 +42,6 @@ def test_jinja_dict(self):
         )
         job_status_0 = JobStatus.from_airflow_dag_run(dag_response.dag_runs[4])
         jinja_dict = job_status_0.jinja_dict
-        print(jinja_dict)
         expected_output = {
             "end_time": datetime(
                 2024, 5, 18, 23, 51, 17, 716003, tzinfo=timezone.utc