diff --git a/tests/client-sdk/safety/conftest.py b/tests/client-sdk/safety/conftest.py
index 9c5ff7352..c4570801c 100644
--- a/tests/client-sdk/safety/conftest.py
+++ b/tests/client-sdk/safety/conftest.py
@@ -7,7 +7,7 @@
 
 def pytest_addoption(parser):
     parser.addoption(
-        "--safety_shield",
+        "--safety-shield",
         action="store",
         default="meta-llama/Llama-Guard-3-1B",
         help="Specify the safety shield model to use for testing",
@@ -18,5 +18,5 @@ def pytest_generate_tests(metafunc):
     if "llama_guard_text_shield_id" in metafunc.fixturenames:
         metafunc.parametrize(
             "llama_guard_text_shield_id",
-            [metafunc.config.getoption("--safety_shield")],
+            [metafunc.config.getoption("--safety-shield")],
         )