diff --git a/aixplain/modules/metadata.py b/aixplain/modules/metadata.py
index f7fac5f5..07007ebe 100644
--- a/aixplain/modules/metadata.py
+++ b/aixplain/modules/metadata.py
@@ -43,6 +43,7 @@ def __init__(
         file_extension: Optional[FileType] = None,
         languages: List[Language] = [],
         dsubtype: DataSubtype = DataSubtype.OTHER,
+        id: Optional[Text] = None,
         **kwargs
     ) -> None:
         """MetaData Class
@@ -62,6 +63,7 @@ def __init__(
             file_extension (Optional[FileType], optional): File extension (e.g. CSV, TXT, etc.). Defaults to None.
             languages (List[Language], optional): List of languages which the data consists of. Defaults to [].
             dsubtype (DataSubtype, optional): Data subtype (e.g., age, topic, race, split, etc.), used in datasets metadata. Defaults to Other.
+            id (Optional[Text], optional): Data ID. Defaults to None.
         """
         self.name = name
         if isinstance(dtype, str):
@@ -91,4 +93,5 @@ def __init__(
                 language = Language(language)
             self.languages.append(language)
         self.dsubtype = dsubtype
+        self.id = id
         self.kwargs = kwargs
diff --git a/tests/functional/pipelines/run_test.py b/tests/functional/pipelines/run_test.py
index e7af6c4e..e4389587 100644
--- a/tests/functional/pipelines/run_test.py
+++ b/tests/functional/pipelines/run_test.py
@@ -109,24 +109,24 @@ def test_run_multipipe_with_datasets(batchmode: bool):
     assert response["status"] == "SUCCESS"
 
 
-@pytest.mark.parametrize("batchmode", [True, False])
-def test_run_segment_reconstruct(batchmode: bool):
+def test_run_segment_reconstruct():
     pipeline = PipelineFactory.list(query="Segmentation/Reconstruction Functional Test - DO NOT DELETE")["results"][0]
-    response = pipeline.run("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", **{"batchmode": batchmode})
+    response = pipeline.run("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav")
 
     assert response["status"] == "SUCCESS"
     output = response["data"][0]
     assert output["label"] == "Output 1"
 
 
-@pytest.mark.parametrize("batchmode", [True, False])
-def test_run_metric(batchmode: bool):
+def test_run_metric():
     pipeline = PipelineFactory.list(query="ASR Metric Functional Test - DO NOT DELETE")["results"][0]
-    response = pipeline.run({
-        "AudioInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav",
-        "ReferenceInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt"
-    }, **{"batchmode": batchmode})
-    
+    response = pipeline.run(
+        {
+            "AudioInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav",
+            "ReferenceInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt",
+        }
+    )
+
     assert response["status"] == "SUCCESS"
     assert len(response["data"]) == 2
     assert response["data"][0]["label"] in ["TranscriptOutput", "ScoreOutput"]
@@ -134,34 +134,30 @@ def test_run_metric(batchmode: bool):
 
 
 @pytest.mark.parametrize(
-    "batchmode,input_data,output_data", 
+    "input_data,output_data",
     [
-        (True, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
-        (False, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
-        (True, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput"),
-        (False, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput")
-    ]
+        ("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
+        ("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput"),
+    ],
 )
-def test_run_router(batchmode: bool, input_data: str, output_data: str):
+def test_run_router(input_data: str, output_data: str):
     pipeline = PipelineFactory.list(query="Router Test - DO NOT DELETE")["results"][0]
-    response = pipeline.run(input_data, **{"batchmode": batchmode})
-    
+    response = pipeline.run(input_data)
+
     assert response["status"] == "SUCCESS"
     assert response["data"][0]["label"] == output_data
 
 
 @pytest.mark.parametrize(
-    "batchmode,input_data,output_data", 
+    "input_data,output_data",
     [
-        (True, "I love it.", "PositiveOutput"),
-        (False, "I love it.", "PositiveOutput"),
-        (True, "I hate it.", "NegativeOutput"),
-        (False, "I hate it.", "NegativeOutput")
-    ]
+        ("I love it.", "PositiveOutput"),
+        ("I hate it.", "NegativeOutput"),
+    ],
 )
-def test_run_decision(batchmode: bool, input_data: str, output_data: str):
+def test_run_decision(input_data: str, output_data: str):
     pipeline = PipelineFactory.list(query="Decision Test - DO NOT DELETE")["results"][0]
-    response = pipeline.run(input_data, **{"batchmode": batchmode})
-    
+    response = pipeline.run(input_data)
+
     assert response["status"] == "SUCCESS"
-    assert response["data"][0]["label"] == output_data
\ No newline at end of file
+    assert response["data"][0]["label"] == output_data