diff --git a/project/CelebornBuild.scala b/project/CelebornBuild.scala
index c45ba1dba..f62c398b7 100644
--- a/project/CelebornBuild.scala
+++ b/project/CelebornBuild.scala
@@ -1203,7 +1203,8 @@ trait FlinkClientProjects {
           flinkStreamingDependency,
           flinkClientsDependency,
           flinkRuntimeWebDependency
-        ) ++ commonUnitTestDependencies
+        ) ++ commonUnitTestDependencies,
+        (Test / envVars) += ("FLINK_VERSION", flinkVersion)
       )
   }