diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
index e8aa06ba245e..42b4700d8687 100644
--- a/tests/python/frontend/onnx/test_forward.py
+++ b/tests/python/frontend/onnx/test_forward.py
@@ -6673,8 +6673,8 @@ def verify_bernoulli_with_ort(
         out_dtype="int32",
         seed=None,
         out_shape=None,
-        target=None,
-        dev=None,
+        target=target,
+        dev=dev,
         use_vm=False,
         opset=None,
         freeze_params=False,