Fixes tree view