diff --git a/ci/run_cugraph_dgl_pytests.sh b/ci/run_cugraph_dgl_pytests.sh
index 83c26a5..e85eb04 100755
--- a/ci/run_cugraph_dgl_pytests.sh
+++ b/ci/run_cugraph_dgl_pytests.sh
@@ -4,6 +4,6 @@
 set -euo pipefail
 
 # Support invoking run_cugraph_dgl_pytests.sh outside the script directory
-cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cugraph-dgl/tests
+cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cugraph-dgl/cugraph_dgl
 
 pytest --cache-clear --ignore=mg "$@" .
diff --git a/ci/test_python.sh b/ci/test_python.sh
index 8950cb7..5e3cfaa 100755
--- a/ci/test_python.sh
+++ b/ci/test_python.sh
@@ -108,8 +108,8 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
     "pylibcugraphops=${RAPIDS_VERSION}" \
     "cugraph=${RAPIDS_VERSION}" \
     "cugraph-pyg=${RAPIDS_VERSION}" \
-    "pytorch::pytorch>=2.3,<2.4" \
-    "ogb"
+    'pytorch::pytorch>=2.3,<2.4' \
+    'ogb'
 
   rapids-print-env
 
@@ -147,8 +147,8 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
     --channel pytorch \
     'mkl<2024.1.0' \
     "pylibwholegraph=${RAPIDS_VERSION}" \
-    "pytorch::pytorch>=2.3,<2.4" \
-    "ogb"
+    'pytorch::pytorch>=2.3,<2.4' \
+    'ogb'
 
   rapids-print-env
 
diff --git a/ci/test_wheel_cugraph-dgl.sh b/ci/test_wheel_cugraph-dgl.sh
index 3eaa098..d709145 100755
--- a/ci/test_wheel_cugraph-dgl.sh
+++ b/ci/test_wheel_cugraph-dgl.sh
@@ -33,4 +33,4 @@ python -m pip install \
     'dgl==2.4.0' \
     'torch>=2.3.0,<2.4'
 
-python -m pytest python/cugraph-dgl/tests
+python -m pytest python/cugraph-dgl/cugraph_dgl/tests