diff --git a/tools/Polygraphy/tests/util/test_array.py b/tools/Polygraphy/tests/util/test_array.py
index eaea1ef8d..a7e7811f0 100644
--- a/tools/Polygraphy/tests/util/test_array.py
+++ b/tools/Polygraphy/tests/util/test_array.py
@@ -96,7 +96,7 @@ def test_is_on_cpu(obj, is_on_cpu):
         (cuda.DeviceArray(shape=(2, 3), dtype=DataType.FLOAT32), True),
     ],
 )
-def test_is_on_cpu(obj, is_on_gpu):
+def test_is_on_gpu(obj, is_on_gpu):
     assert util.array.is_on_gpu(obj) == is_on_gpu