diff --git a/tests/test_distributed/test_with_dask/test_with_dask.py b/tests/test_distributed/test_with_dask/test_with_dask.py
index 77db640c2a78..53e263b5e06e 100644
--- a/tests/test_distributed/test_with_dask/test_with_dask.py
+++ b/tests/test_distributed/test_with_dask/test_with_dask.py
@@ -1633,7 +1633,7 @@ def test_feature_weights(self, client: "Client") -> None:
         for i in range(kCols):
             fw[i] *= float(i)
         fw = da.from_array(fw)
-        parser = os.path.join(tm.demo_dir(__file__), "json-model", "json_parser.py")
+        parser = os.path.join(tm.demo_dir(__file__), "guide-python", "model_parser.py")
         poly_increasing = get_feature_weights(
             X=X,
             y=y,