diff --git a/src/Ursa/Common/VisualHelpers.cs b/src/Ursa/Common/VisualHelpers.cs
index 37195ac..511a3d2 100644
--- a/src/Ursa/Common/VisualHelpers.cs
+++ b/src/Ursa/Common/VisualHelpers.cs
@@ -1,13 +1,14 @@
 using Avalonia;
+using Avalonia.Controls;
 using Avalonia.VisualTree;
 
 namespace Ursa.Common;
 
 public static class VisualHelpers
 {
-    public static T? GetContainerFromEventSource<T>(this Visual? source)
+    public static T? GetContainerFromEventSource<T>(this Visual? source) where T: Control
     {
-        var item = source.GetSelfAndVisualAncestors().OfType<T>().FirstOrDefault();
+        var item = source?.GetSelfAndVisualAncestors().OfType<T>().FirstOrDefault();
         return item;
     }
 }
\ No newline at end of file