diff --git a/core/src/main/scala/org/apache/spark/ui/jobs/StagePage.scala b/core/src/main/scala/org/apache/spark/ui/jobs/StagePage.scala
index b83a49f79c8a..432c3fbfbce1 100644
--- a/core/src/main/scala/org/apache/spark/ui/jobs/StagePage.scala
+++ b/core/src/main/scala/org/apache/spark/ui/jobs/StagePage.scala
@@ -659,8 +659,12 @@ private[ui] class StagePage(parent: StagesTab) extends WebUIPage("stage") {
       {TIMELINE_LEGEND}
     </div> ++
     <script type="text/javascript">
-      {Unparsed(s"drawTaskAssignmentTimeline(" +
-      s"$groupArrayStr, $executorsArrayStr, $minLaunchTime, $maxFinishTime)")}
+      {
+        Unparsed(s"""$$(function() {
+          drawTaskAssignmentTimeline(
+            $groupArrayStr, $executorsArrayStr, $minLaunchTime, $maxFinishTime)
+        })""")
+      }
     </script>
   }