TreeView - Selection state inconsistency between ContextMenu and ContextFlyout implementations

[Coloris]

Describe the bug

When using a TreeView with ContextFlyout as a context menu, the selection state / focus of TreeView items is lost after interacting with the MenuFlyout, Subsequent item selection requires two clicks instead of the expected single click behavior.

However, when using the traditional ContextMenu, the selection state is properly maintained.

To Reproduce

Working example with ContextMenu

<TreeView ItemsSource="{Binding Items}">
	<TreeView.ItemTemplate>
		<TreeDataTemplate ItemsSource="{Binding Children}">
			<TextBlock Text="{Binding Header}" Background="Transparent">
				<TextBlock.ContextMenu>
					<ContextMenu>
						<MenuItem Header="{Binding Header}"/>
					</ContextMenu>
				</TextBlock.ContextMenu>
			</TextBlock>
		</TreeDataTemplate>
	</TreeView.ItemTemplate>
</TreeView>

Issue example with MenuFlyout

<TreeView ItemsSource="{Binding Items}">
	<TreeView.ItemTemplate>
		<TreeDataTemplate ItemsSource="{Binding Children}">
			<TextBlock Text="{Binding Header}" Background="Transparent">
				<TextBlock.ContextFlyout>
					<MenuFlyout ShowMode="TransientWithDismissOnPointerMoveAway">
						<MenuItem Header="{Binding Header}"/>
					</MenuFlyout>
				</TextBlock.ContextFlyout>
			</TextBlock>
		</TreeDataTemplate>
	</TreeView.ItemTemplate>
</TreeView>

Expected behavior

ContextFlyout should maintain consistent selection behavior like the existing ContextMenu implementation when used in a TreeView control

Avalonia version

11.2.2

OS

Windows 11 [Version 10.0.26100.2454]

Additional context

No response