diff --git a/Material.Avalonia.Demo/Material.Avalonia.Demo.csproj b/Material.Avalonia.Demo/Material.Avalonia.Demo.csproj
index 49bb2937..4275fdcb 100644
--- a/Material.Avalonia.Demo/Material.Avalonia.Demo.csproj
+++ b/Material.Avalonia.Demo/Material.Avalonia.Demo.csproj
@@ -24,4 +24,10 @@
   <ItemGroup>
     <AvaloniaResource Include="Assets\**"/>
   </ItemGroup>
+
+  <ItemGroup>
+    <Compile Update="Styles\ShowMeTheXamlStyles.XamlDisplayAvaloniaEditThemeBehavior.cs">
+      <DependentUpon>ShowMeTheXamlStyles.axaml</DependentUpon>
+    </Compile>
+  </ItemGroup>
 </Project>
diff --git a/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.XamlDisplayAvaloniaEditThemeBehavior.cs b/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.XamlDisplayAvaloniaEditThemeBehavior.cs
new file mode 100644
index 00000000..47737daa
--- /dev/null
+++ b/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.XamlDisplayAvaloniaEditThemeBehavior.cs
@@ -0,0 +1,20 @@
+using System.Runtime.InteropServices;
+using ShowMeTheXaml.Avalonia.AvaloniaEdit;
+
+namespace Material.Avalonia.Demo.Styles;
+
+public class CustomXamlDisplayAvaloniaEditThemeBehavior : XamlDisplayAvaloniaEditThemeBehavior {
+    protected override void OnAttachedToVisualTree() {
+        if (RuntimeInformation.ProcessArchitecture == Architecture.Wasm) {
+            return;
+        }
+        base.OnAttachedToVisualTree();
+    }
+
+    protected override void OnDetachedFromVisualTree() {
+        if (RuntimeInformation.ProcessArchitecture == Architecture.Wasm) {
+            return;
+        }
+        base.OnDetachedFromVisualTree();
+    }
+}
\ No newline at end of file
diff --git a/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.axaml b/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.axaml
index 174de87e..edf04a08 100644
--- a/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.axaml
+++ b/Material.Avalonia.Demo/Styles/ShowMeTheXamlStyles.axaml
@@ -3,7 +3,8 @@
         xmlns:avaloniaEdit="https://github.com/avaloniaui/avaloniaedit"
         xmlns:avaloniaEdit1="clr-namespace:ShowMeTheXaml.Avalonia.AvaloniaEdit;assembly=ShowMeTheXaml.Avalonia.AvaloniaEdit"
         xmlns:showMeTheXaml="clr-namespace:ShowMeTheXaml;assembly=ShowMeTheXaml.Avalonia"
-        xmlns:controls="clr-namespace:Material.Styles.Controls;assembly=Material.Styles">
+        xmlns:controls="clr-namespace:Material.Styles.Controls;assembly=Material.Styles"
+        xmlns:styles="clr-namespace:Material.Avalonia.Demo.Styles">
   <StyleInclude Source="avares://ShowMeTheXaml.Avalonia.AvaloniaEdit/XamlDisplayStyles.axaml" />
 
   <Style Selector="showMeTheXaml|XamlDisplay">
@@ -38,7 +39,7 @@
                                        Background="{Binding $parent[showMeTheXaml:XamlDisplay].Background}">
                 <Interaction.Behaviors>
                   <avaloniaEdit1:CustomizeEditorBehavior />
-                  <avaloniaEdit1:XamlDisplayAvaloniaEditThemeBehavior />
+                  <styles:CustomXamlDisplayAvaloniaEditThemeBehavior />
                 </Interaction.Behaviors>
               </avaloniaEdit:TextEditor>
             </ScrollViewer>
@@ -98,7 +99,7 @@
                 <avaloniaEdit1:CustomizeEditorBehavior />
                 <avaloniaEdit1:XamlDisplayAvaloniaEditTextBindingBehavior
                   MarkupTextEditor="{Binding ElementName=MarkupTextEditor}" />
-                <avaloniaEdit1:XamlDisplayAvaloniaEditThemeBehavior />
+                <styles:CustomXamlDisplayAvaloniaEditThemeBehavior />
               </Interaction.Behaviors>
             </avaloniaEdit:TextEditor>
           </ScrollViewer>