diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py
index 68ad93682ed18..19b31d6fd3ed4 100644
--- a/tests/python/frontend/mxnet/test_forward.py
+++ b/tests/python/frontend/mxnet/test_forward.py
@@ -1936,7 +1936,7 @@ def verify(
             ):
                 target += " -libs=thrust"
             for kind in ["graph", "debug"]:
-                intrp = relay.create_executor(kind, mod=mod, dev=dev, target=target)
+                intrp = relay.create_executor(kind, mod=mod, device=dev, target=target)
                 op_res = intrp.evaluate()(data)
                 tvm.testing.assert_allclose(
                     op_res.asnumpy(), ref_res.asnumpy(), rtol=1e-3, atol=1e-5
diff --git a/tests/scripts/task_python_docs.sh b/tests/scripts/task_python_docs.sh
index 459b680daeb1b..285e5c983d2bb 100755
--- a/tests/scripts/task_python_docs.sh
+++ b/tests/scripts/task_python_docs.sh
@@ -51,6 +51,10 @@ find . -type f -path "*.log" | xargs rm -f
 find . -type f -path "*.pyc" | xargs rm -f
 make cython3
 
+# cleanup stale states for rust
+rm rust/tvm-sys/src/c_runtime_api.rs
+rm -rf rust/target/doc
+
 cd docs
 PYTHONPATH=`pwd`/../python make html |& tee /tmp/$$.log.txt
 if grep -E "failed to execute|Segmentation fault" < /tmp/$$.log.txt; then