diff --git a/aws/logs_monitoring/tests/test_awslogs_handler.py b/aws/logs_monitoring/tests/test_awslogs_handler.py
index 49f5d824..65b0ab59 100644
--- a/aws/logs_monitoring/tests/test_awslogs_handler.py
+++ b/aws/logs_monitoring/tests/test_awslogs_handler.py
@@ -123,6 +123,10 @@ def test_awslogs_handler_step_functions_tags_added_properly(
         self.assertEqual(
             awslogs_handler.metadata[DD_SOURCE], AwsEventSource.STEPFUNCTION.value
         )
+        self.assertEqual(
+            awslogs_handler.metadata[DD_HOST],
+            "arn:aws:states:us-east-1:12345678910:stateMachine:StepFunction1",
+        )
 
         # SF customized log group
         eventFromCustomizedLogGroup = {
@@ -158,7 +162,7 @@ def test_awslogs_handler_step_functions_tags_added_properly(
         )
         self.assertEqual(
             awslogs_handler.metadata[DD_HOST],
-            "arn:aws:states:us-east-1:12345678910:statemachine:stepfunction1",
+            "arn:aws:states:us-east-1:12345678910:stateMachine:StepFunction1",
         )
 
         eventFromCustomizedLogGroup = {