Fix findPsiElements.
Fix if uml.
Optimize if uml.