diff --git a/localstack-core/localstack/services/stepfunctions/provider.py b/localstack-core/localstack/services/stepfunctions/provider.py
index 777573d9d769d..0399d32890ac6 100644
--- a/localstack-core/localstack/services/stepfunctions/provider.py
+++ b/localstack-core/localstack/services/stepfunctions/provider.py
@@ -1143,10 +1143,11 @@ def test_state(
         self,
         context: RequestContext,
         definition: Definition,
-        role_arn: Arn,
+        role_arn: Arn = None,
         input: SensitiveData = None,
         inspection_level: InspectionLevel = None,
         reveal_secrets: RevealSecrets = None,
+        variables: SensitiveData = None,
         **kwargs,
     ) -> TestStateOutput:
         StepFunctionsProvider._validate_definition(