diff --git a/app/src/components/ChartPanel/Chart.tsx b/app/src/components/ChartPanel/Chart.tsx
index bfcb557..cf9d96e 100644
--- a/app/src/components/ChartPanel/Chart.tsx
+++ b/app/src/components/ChartPanel/Chart.tsx
@@ -81,7 +81,7 @@ function Chart(props: Props) {
 
 const mapStateToProps = (state: AppState) => {
   return {
-    tree: state.tree.get('tree'),
+    tree: state.connection.tree,
   }
 }