Fix Tree UI Bug

[pogi7]

Checklist before submitting a bug report

• [x] I have reviewed CONTRIBUTING.md document
• [x] I have done a quick search on the issue tracker to check if the bug report is not redundant

Context

Fix bugs address in the attachment

user_omlvision.zip

Steps to reproduce the bug

Look at the attachment on how to address bugs

Current result

Look at the attachment on the current result

Expected result

The bugs shouldn't occur

Additional context

Look at the attachment

Possible fix

[Not required but you can suggest possible fixes for the bug or link a place in the code that could possibly be causing the bug]

Technical data

• Operating system: [What is the OS]
• Version of the repo: [What is the release version or a commit of the repo]
• VSCode version: [What is the VSCode version]

/label ~"bug"