diff --git a/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py b/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py
index cfa0a8b65d45..0be1b50231eb 100644
--- a/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py
+++ b/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py
@@ -13,7 +13,7 @@
 # This test can also be performed for 1 node, but it takes a very long time, and CodeBuild job may expire before the
 # test ends.
 @pytest.mark.parametrize("num_nodes", [4], indirect=True)
-def test_mxnet_sagemaker_training_performance(mxnet_training, num_nodes, region, gpu_only):
+def test_mxnet_sagemaker_training_performance(mxnet_training, num_nodes, region, gpu_only, py3_only):
     """
     Run MX sagemaker training performance test