When TreeView is placed inside floating VS ToolWindow and item is selected the visual state remains unchanged after focus leaves the TreeView.
The XAML team has recently reviewed this issue and it is no longer reproducible. If you are still experiencing it and have a sample runnable project reproducing it. please contact us through the support ticketing system with details on your setup and a reference to this item.