diff --git a/templates/jax/0.4.30/jax.arm.py39.cpu.config b/templates/jax/0.4.30/jax.arm.py39.cpu.config
index 0a6a314aa78709d71ee546fea49b14126de4f37f..2361783f1a5941580fed61aa7bea47b792c36388 100644
--- a/templates/jax/0.4.30/jax.arm.py39.cpu.config
+++ b/templates/jax/0.4.30/jax.arm.py39.cpu.config
@@ -9,16 +9,19 @@ module purge
 ./jarvis -install hpckit/${HPCKIT_VERSION} any
 module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles
 module load bisheng/compiler${BISHENG_VERSION}/bishengmodule
-
+module load bisheng/hmpi${HPCKIT_VERSION}/release
 ./jarvis -install python3/3.9.9 bisheng
+module use ./software/moduledeps/bisheng${BISHENG_VERSION}
 module load python3/3.9.9
+rm -rf ${JARVIS_TMP_DOWNLOAD}/jax-v0.4.30
 tar -zxvf ${JARVIS_DOWNLOAD}/jax-v0.4.30.tar.gz -C ${JARVIS_TMP_DOWNLOAD}
 
 [ENV]
 module purge
 module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles
 module load bisheng/compiler${BISHENG_VERSION}/bishengmodule
-
+module load bisheng/hmpi${HPCKIT_VERSION}/release
+module use ./software/moduledeps/bisheng${BISHENG_VERSION}
 module load python3/3.9.9
 
 [APP]
@@ -38,3 +41,4 @@ pip3 uninstall jax -y
 run = python3
 binary = jax_jit_test.py
 nodes = 1
+