Fixes small bug when switching from treemap to table