diff --git a/tests/test_bump_schema_versions.py b/tests/test_bump_schema_versions.py
index 59ee50cf8..13d3c19ee 100644
--- a/tests/test_bump_schema_versions.py
+++ b/tests/test_bump_schema_versions.py
@@ -51,6 +51,7 @@ def test_get_list_of_incremented_versions(self, mock_get_schema: MagicMock):
         new_session_version = str(Version.parse(old_session_version).bump_patch())
 
         def side_effect(model):
+            """Side effect for mock_get_schema"""
             if model == Subject:
                 return {"properties": {"schema_version": {"default": old_subject_version}}}
             elif model == Session:
@@ -151,6 +152,7 @@ def test_run_job(
         mock_get_list_of_models.return_value = [Subject, Session]
 
         def side_effect(model):
+            """Return values for get_schema_json"""
             if model == Subject:
                 return {"properties": {"schema_version": {"default": old_subject_version}}}
             elif model == Session: