diff --git a/spark-plugin/plugin/src/main/scala/org/apache/spark/dataflint/DataflintSparkUILoader.scala b/spark-plugin/plugin/src/main/scala/org/apache/spark/dataflint/DataflintSparkUILoader.scala
index 59ed8e1..e9b4dd5 100644
--- a/spark-plugin/plugin/src/main/scala/org/apache/spark/dataflint/DataflintSparkUILoader.scala
+++ b/spark-plugin/plugin/src/main/scala/org/apache/spark/dataflint/DataflintSparkUILoader.scala
@@ -40,4 +40,8 @@ object DataflintSparkUILoader {
   def install(context: SparkContext): String = {
     new DataflintSparkUIInstaller().install(context)
   }
+
+  def loadUI(ui: SparkUI): String = {
+    new DataflintSparkUIInstaller().loadUI(ui)
+  }
 }