From f2f20323ca03ecdb21d77de5c35b6270417bfd1e Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Sun, 3 Dec 2023 15:07:29 -0500
Subject: [PATCH] Use base type property

---
 .../ViewModels/Inference/StackEditableCardViewModel.cs        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix.Avalonia/ViewModels/Inference/StackEditableCardViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/Inference/StackEditableCardViewModel.cs
index 7f2a1d35..d26b646c 100644
--- a/StabilityMatrix.Avalonia/ViewModels/Inference/StackEditableCardViewModel.cs
+++ b/StabilityMatrix.Avalonia/ViewModels/Inference/StackEditableCardViewModel.cs
@@ -62,7 +62,7 @@ public partial class StackEditableCardViewModel : StackViewModelBase
     partial void OnIsEditEnabledChanged(bool value)
     {
         // Propagate edit state to children
-        foreach (var module in Cards.OfType<ModuleBase>())
+        foreach (var module in Cards.OfType<StackExpanderViewModel>())
         {
             module.IsEditEnabled = value;
         }
@@ -73,7 +73,7 @@ public partial class StackEditableCardViewModel : StackViewModelBase
     {
         base.OnCardAdded(item);
 
-        if (item is ModuleBase module)
+        if (item is StackExpanderViewModel module)
         {
             // Inherit our edit state
             module.IsEditEnabled = IsEditEnabled;