diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh
index 2b7c982cb..d2b495faf 100755
--- a/scripts/ci/cuda/ci_install_dependency.sh
+++ b/scripts/ci/cuda/ci_install_dependency.sh
@@ -298,7 +298,7 @@ if [ "$CU_VERSION" = "cu130" ]; then
 else
     NVRTC_SPEC="nvidia-cuda-nvrtc-cu12"
 fi
-$PIP_CMD install mooncake-transfer-engine==0.3.9 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
+$PIP_CMD install mooncake-transfer-engine==0.3.10 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
 
 # Install other test dependencies
 if [ "$IS_BLACKWELL" != "1" ]; then
diff --git a/test/registered/ep/test_mooncake_ep_small.py b/test/registered/ep/test_mooncake_ep_small.py
index a59a44c31..ce87eed14 100644
--- a/test/registered/ep/test_mooncake_ep_small.py
+++ b/test/registered/ep/test_mooncake_ep_small.py
@@ -20,7 +20,6 @@ register_cuda_ci(est_time=660, suite="stage-c-test-deepep-4-gpu-h100")
 ib_devices = get_rdma_devices_args()
 
 
-@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
 class TestTP(CustomTestCase):
     extra_args = []
 
@@ -84,7 +83,6 @@ class TestTP(CustomTestCase):
         self.assertGreater(metrics["accuracy"], 0.60)
 
 
-@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
 class TestPureDP(TestTP):
     extra_args = [
         "--enable-dp-attention",