diff --git a/src/Perspex.Controls/TreeView.cs b/src/Perspex.Controls/TreeView.cs
index 0a83891bc8..ea2b5d632c 100644
--- a/src/Perspex.Controls/TreeView.cs
+++ b/src/Perspex.Controls/TreeView.cs
@@ -197,15 +197,18 @@ namespace Perspex.Controls
         /// <param name="selected">Whether the control is selected</param>
         private void MarkContainerSelected(IControl container, bool selected)
         {
-            var selectable = container as ISelectable;
-
-            if (selectable != null)
-            {
-                selectable.IsSelected = selected;
-            }
-            else
+            if (container != null)
             {
-                ((IPseudoClasses)container.Classes).Set(":selected", selected);
+                var selectable = container as ISelectable;
+
+                if (selectable != null)
+                {
+                    selectable.IsSelected = selected;
+                }
+                else
+                {
+                    ((IPseudoClasses)container.Classes).Set(":selected", selected);
+                }
             }
         }
     }