From 52c5efb9ed14c4eaaf237ba98fd3735240b6ca63 Mon Sep 17 00:00:00 2001
From: Kent Yao <yaooqinn@hotmail.com>
Date: Wed, 16 May 2018 18:58:23 +0800
Subject: [PATCH] explicitly visit spark env 3

---
 src/main/scala/org/apache/spark/KyuubiSparkUtil.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala b/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala
index 6c9cb2684..18efe5067 100644
--- a/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala
+++ b/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala
@@ -79,7 +79,7 @@ object KyuubiSparkUtil extends Logging {
     val classLoader = new KyuubiFirstClassLoader(Array(url), getContextOrSparkClassLoader())
     val envCls = classLoader.loadClass("org.apache.spark.SparkEnv", true)
     val envObj = classLoader.loadClass("org.apache.spark.SparkEnv$", true)
-    ReflectUtils.invokeStaticMethod(envObj, "set", Seq(envCls), null)
+    ReflectUtils.invokeStaticMethod(envCls, "set", Seq(envCls), null)
     classLoader
   }