diff --git a/dev-python/pyrefly/pyrefly-1.1.0.ebuild b/dev-python/pyrefly/pyrefly-1.1.0.ebuild
index 00438a278b..d7adb9e545 100644
--- a/dev-python/pyrefly/pyrefly-1.1.0.ebuild
+++ b/dev-python/pyrefly/pyrefly-1.1.0.ebuild
@@ -473,5 +473,8 @@ src_test() {
 	local -x REPORT_TEST_PATH="${S}/lib/test/report/test_files"
 	local -x STUBGEN_TEST_PATH="${S}/lib/test/stubgen"
 
+	local -x COVERAGE_TEST_PATH="${S}/lib/test/coverage/test_files"
+	local -x SHAPE_DSL_TEST_PATH="${WORKDIR}/${P}/tensor-shapes"
+
 	cargo_src_test --no-fail-fast
 }