better way of making sure Treemap is displayed on actions