diff --git a/FModel/Extensions/VisualTreeExtensions.cs b/FModel/Extensions/VisualTreeExtensions.cs
index c898336f..1a54a808 100644
--- a/FModel/Extensions/VisualTreeExtensions.cs
+++ b/FModel/Extensions/VisualTreeExtensions.cs
@@ -11,7 +11,9 @@ public static class VisualTreeExtensions
         {
             if (current is T t)
                 return t;
-            current = VisualTreeHelper.GetParent(current);
+            current = current is FrameworkContentElement contentElement
+                ? contentElement.Parent
+                : VisualTreeHelper.GetParent(current);
         }
         return null;
     }