diff --git a/tests/test_agg_mds_tcia_adapter.py b/tests/test_agg_mds_tcia_adapter.py
index 56c11f4..49a42e2 100644
--- a/tests/test_agg_mds_tcia_adapter.py
+++ b/tests/test_agg_mds_tcia_adapter.py
@@ -176,7 +176,7 @@ def test_get_metadata_tcia():
                 "commons": "TCIA",
                 "study_title": "",
                 "program_name": "Collection1",
-                "image_count": "10",
+                "image_count": 10,
                 "description": "TCIA data from collection: Collection1.",
                 "tags": [{"category": "program_name", "name": "Collection1"}],
             },
@@ -189,7 +189,7 @@ def test_get_metadata_tcia():
                 "commons": "TCIA",
                 "study_title": "",
                 "program_name": "Collection2",
-                "image_count": "20",
+                "image_count": 20,
                 "description": "TCIA data from collection: Collection2.",
                 "tags": [{"category": "program_name", "name": "Collection2"}],
             },