diff --git a/test/test_artifacts/v2/scripts/run_pysdk_tests.sh b/test/test_artifacts/v2/scripts/run_pysdk_tests.sh
index 912fdf91..2500585a 100644
--- a/test/test_artifacts/v2/scripts/run_pysdk_tests.sh
+++ b/test/test_artifacts/v2/scripts/run_pysdk_tests.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
 
 # We need to checkout the version of sagemaker-python-sdk that is installed in the mamba environment.
-pysdk_version=$(micromamba list | grep sagemaker-python-sdk | tr -s ' ' | cut -d ' ' -f 3)
 
+pysdk_version=$(micromamba list | grep sagemaker-python-sdk | tr -s ' ' | cut -d ' ' -f 3)
 # Checkout the corresponding sagemaker-python-sdk version
 git checkout tags/v$pysdk_version