From 89e8d4ea98c8e640dd013418c6891d366f376fe9 Mon Sep 17 00:00:00 2001
From: Tal Bereznitskey <berzniz@gmail.com>
Date: Fri, 27 Aug 2021 13:07:38 +0300
Subject: [PATCH] Bug fix: tree not always loading on navigation

---
 src/js/index.jsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/js/index.jsx b/src/js/index.jsx
index 7d12915..ec9d15a 100644
--- a/src/js/index.jsx
+++ b/src/js/index.jsx
@@ -11,8 +11,10 @@ let observer
 const observe = () => {
   observer && observer.disconnect()
   const pjaxContainer = document.querySelector('[data-pjax-container]')
+  const pjaxContentContainer = document.querySelector('#repo-content-pjax-container')
   observer = new MutationObserver(start)
-  observer.observe(pjaxContainer, { childList: true })
+  pjaxContainer && observer.observe(pjaxContainer, { childList: true })
+  pjaxContentContainer && observer.observe(pjaxContentContainer, { childList: true })
 }
 
 class Top extends React.Component {