diff --git a/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala b/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala
index db94422d5..41e38f634 100644
--- a/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala
+++ b/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala
@@ -128,7 +128,7 @@ class Ray(override val uid: String) extends SQLAlg with VersionCompatibility wit
 
         tempdf.rdd.mapPartitions { iter =>
 
-          val host: String = if (SparkEnv.get == null || MLSQLSparkUtils.rpcEnv().address == null) {
+          val host: String = if (SparkEnv.get == null || MLSQLSparkUtils.blockManager == null || MLSQLSparkUtils.blockManager.blockManagerId == null) {
             WriteLog.write(List("Ray: Cannot get MLSQLSparkUtils.rpcEnv().address, using NetTool.localHostName()").iterator,
               Map("PY_EXECUTE_USER" -> owner))
             NetTool.localHostName()