diff --git a/samples/NotificationFlyoutSample.Host/NotificationFlyoutSample.Host.csproj b/samples/NotificationFlyoutSample.Host/NotificationFlyoutSample.Host.csproj
index 0dc79d1..0b69427 100644
--- a/samples/NotificationFlyoutSample.Host/NotificationFlyoutSample.Host.csproj
+++ b/samples/NotificationFlyoutSample.Host/NotificationFlyoutSample.Host.csproj
@@ -6,6 +6,7 @@
     <UseWPF>true</UseWPF>
     <Platforms>AnyCPU;x64</Platforms>
     <StartupObject>NotificationFlyoutSample.Host.Program</StartupObject>
+    <AssetTargetFallback>uap10.0.19041</AssetTargetFallback>
   </PropertyGroup>
 
   <ItemGroup>
diff --git a/samples/NotificationFlyoutSample/NotificationFlyoutSample.csproj b/samples/NotificationFlyoutSample/NotificationFlyoutSample.csproj
index 4d99e11..76ef38b 100644
--- a/samples/NotificationFlyoutSample/NotificationFlyoutSample.csproj
+++ b/samples/NotificationFlyoutSample/NotificationFlyoutSample.csproj
@@ -172,14 +172,14 @@
     </Page>
   </ItemGroup>
   <ItemGroup>
-    <ProjectReference Include="..\..\src\TheXamlGuy.NotificationFlyout.Shared.UI\TheXamlGuy.NotificationFlyout.Shared.UI.csproj">
-      <Project>{7d6a151b-c021-4d34-adc1-0171a2f65a00}</Project>
-      <Name>TheXamlGuy.NotificationFlyout.Shared.UI</Name>
-    </ProjectReference>
     <ProjectReference Include="..\..\src\TheXamlGuy.NotificationFlyout.Uwp.UI.Controls\TheXamlGuy.NotificationFlyout.Uwp.UI.Controls.csproj">
       <Project>{8501316d-f994-4a29-9ec4-dc6c6b93f565}</Project>
       <Name>TheXamlGuy.NotificationFlyout.Uwp.UI.Controls</Name>
     </ProjectReference>
+    <ProjectReference Include="..\..\src\TheXamlGuy.NotificationFlyout.Shared.UI\TheXamlGuy.NotificationFlyout.Shared.UI.csproj">
+      <Project>{8501316d-f994-4a29-9ec4-dc6c6b93f565}</Project>
+      <Name>TheXamlGuy.NotificationFlyout.Shared.UI</Name>
+    </ProjectReference>
   </ItemGroup>
   <PropertyGroup Condition=" '$(VisualStudioVersion)' == '' or '$(VisualStudioVersion)' &lt; '14.0' ">
     <VisualStudioVersion>14.0</VisualStudioVersion>
diff --git a/samples/NotificationFlyoutSample/Shell.xaml b/samples/NotificationFlyoutSample/Shell.xaml
index fcace13..f386b57 100644
--- a/samples/NotificationFlyoutSample/Shell.xaml
+++ b/samples/NotificationFlyoutSample/Shell.xaml
@@ -3,9 +3,7 @@
     xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
     xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
     xmlns:controls="using:TheXamlGuy.NotificationFlyout.Uwp.UI.Controls"
-    xmlns:muxc="using:Microsoft.UI.Xaml.Controls"
-    IconSource="/Assets/Icon.ico"
-    LightIconSource="/Assets/Icon-Light.ico">
+    xmlns:muxc="using:Microsoft.UI.Xaml.Controls">
     <controls:NotificationFlyout.FlyoutPresenterStyle>
         <Style TargetType="controls:NotificationFlyoutPresenter">
             <Setter Property="CornerRadius" Value="8" />
diff --git a/samples/NotificationFlyoutSample/Shell.xaml.cs b/samples/NotificationFlyoutSample/Shell.xaml.cs
index 1de9101..f7e0b73 100644
--- a/samples/NotificationFlyoutSample/Shell.xaml.cs
+++ b/samples/NotificationFlyoutSample/Shell.xaml.cs
@@ -1,6 +1,4 @@
-using Windows.UI.Xaml.Controls;
-
-namespace NotificationFlyoutSample
+namespace NotificationFlyoutSample
 {
     public sealed partial class Shell
     {
diff --git a/src/TheXamlGuy.NotificationFlyout.Shared.UI/TheXamlGuy.NotificationFlyout.Shared.UI.csproj b/src/TheXamlGuy.NotificationFlyout.Shared.UI/TheXamlGuy.NotificationFlyout.Shared.UI.csproj
index 79a48cf..6c616c9 100644
--- a/src/TheXamlGuy.NotificationFlyout.Shared.UI/TheXamlGuy.NotificationFlyout.Shared.UI.csproj
+++ b/src/TheXamlGuy.NotificationFlyout.Shared.UI/TheXamlGuy.NotificationFlyout.Shared.UI.csproj
@@ -21,4 +21,9 @@
     <PackageReference Include="System.Drawing.Common" Version="6.0.0-preview.1.21102.12" />
     <PackageReference Include="System.Memory" Version="4.5.4" />
   </ItemGroup>
+
+  <PropertyGroup>
+    <EnableTypeInfoReflection>false</EnableTypeInfoReflection>
+    <EnableXBindDiagnostics>false</EnableXBindDiagnostics>
+  </PropertyGroup>
 </Project>
diff --git a/src/TheXamlGuy.NotificationFlyout.Uwp.UI.Controls/NotificationFlyoutHost/NotificationFlyoutHost.cs b/src/TheXamlGuy.NotificationFlyout.Uwp.UI.Controls/NotificationFlyoutHost/NotificationFlyoutHost.cs
index e2ca9d6..efb4a54 100644
--- a/src/TheXamlGuy.NotificationFlyout.Uwp.UI.Controls/NotificationFlyoutHost/NotificationFlyoutHost.cs
+++ b/src/TheXamlGuy.NotificationFlyout.Uwp.UI.Controls/NotificationFlyoutHost/NotificationFlyoutHost.cs
@@ -46,11 +46,6 @@ namespace TheXamlGuy.NotificationFlyout.Uwp.UI.Controls
             flyout.Hide();
         }
 
-        internal void UpdateThemeVisualState()
-        {
-
-        }
-
         public void SetFlyoutPlacement(string placement)
         {
             if (!_isLoaded)
@@ -73,7 +68,7 @@ namespace TheXamlGuy.NotificationFlyout.Uwp.UI.Controls
             });
         }
 
-        internal void SetOwningFlyout(NotificationFlyout flyout)
+        public void SetOwningFlyout(NotificationFlyout flyout)
         {
             _notificationFlyout = flyout;
 
@@ -103,12 +98,8 @@ namespace TheXamlGuy.NotificationFlyout.Uwp.UI.Controls
                 });
         }
 
-        private NotificationFlyoutPresenter _flyoutPresenter;
-
         protected override void OnApplyTemplate()
         {
-            _flyoutPresenter = GetTemplateChild("FlyoutPresenter") as NotificationFlyoutPresenter;
-
             _flyout = GetTemplateChild("Flyout") as Flyout;
             if (_flyout != null)
             {
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-default.ico b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-default.ico
new file mode 100644
index 0000000..f69c656
Binary files /dev/null and b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-default.ico differ
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-light.ico b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-light.ico
new file mode 100644
index 0000000..f66fb7e
Binary files /dev/null and b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/Assets/notification-icon-light.ico differ
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/NotificationFlyout/NotificationFlyoutXamlHost.cs b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/NotificationFlyout/NotificationFlyoutXamlHost.cs
index b54fda3..cd4a1af 100644
--- a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/NotificationFlyout/NotificationFlyoutXamlHost.cs
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/NotificationFlyout/NotificationFlyoutXamlHost.cs
@@ -5,6 +5,9 @@ using TheXamlGuy.NotificationFlyout.Wpf.UI.Extensions;
 using System;
 using System.Windows;
 using Windows.UI.Xaml.Controls.Primitives;
+using Windows.UI.Xaml.Controls;
+using System.Windows.Media;
+using System.Windows.Media.Imaging;
 
 namespace TheXamlGuy.NotificationFlyout.Wpf.UI.Controls
 {
@@ -105,20 +108,7 @@ namespace TheXamlGuy.NotificationFlyout.Wpf.UI.Controls
 
         private void OnTaskbarChanged(object sender, EventArgs args) => UpdateWindow();
 
-        private void OnThemeChanged(object sender, SystemPersonalisationChangedEventArgs args)
-        {
-            UpdateFlyoutTheme(args.IsColorPrevalence);
-            UpdateIcons();
-        }
-
-        private void UpdateFlyoutTheme(bool isColorPrevalence)
-        {
-            var content = GetHostContent();
-            if (content != null)
-            {
-             //   content.UpdateFlyoutTheme(isColorPrevalence);
-            }
-        }
+        private void OnThemeChanged(object sender, SystemPersonalisationChangedEventArgs args) => UpdateIcons();
 
         private void PrepareContextMenu()
         {
@@ -168,18 +158,23 @@ namespace TheXamlGuy.NotificationFlyout.Wpf.UI.Controls
             if (!IsLoaded) return;
             if (_flyout == null) return;
 
-            var iconSource = _flyout.IconSource;
-            var lightIconSource = _flyout.LightIconSource;
-
             var shellTrayHandle = WindowHelper.GetHandle(ShellTrayHandleName);
             if (shellTrayHandle == null) return;
 
-            var desiredIconSource = _systemPersonalisationHelper.Theme == SystemTheme.Dark ? iconSource : lightIconSource;
-            if (desiredIconSource == null) return;
-
             var dpi = WindowHelper.GetDpi(shellTrayHandle);
-            using var icon = await desiredIconSource.ConvertToIconAsync(dpi);
-            _notificationIconHelper.SetIcon(icon.Handle);
+
+            var desiredIconSource = _systemPersonalisationHelper.Theme == SystemTheme.Dark ? _flyout.IconSource : _flyout.LightIconSource;
+            if (desiredIconSource == null)
+            {
+                var fallbackIconSource = new BitmapImage(new Uri($"pack://application:,,,/{GetType().Namespace};component/Assets/notification-icon-{(_systemPersonalisationHelper.Theme == SystemTheme.Dark ? "default" : "light")}.ico"));
+                using var icon = fallbackIconSource.ConvertToIcon(dpi);
+                _notificationIconHelper.SetIcon(icon.Handle);
+            }
+            else
+            {
+                using var icon = await desiredIconSource.ConvertToIconAsync(dpi);
+                _notificationIconHelper.SetIcon(icon.Handle);
+            }
         }
 
         private void UpdateWindow()
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls.csproj b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls.csproj
index 59c1384..861b922 100644
--- a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls.csproj
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls.csproj
@@ -12,6 +12,11 @@
     <Version>1.0.0</Version>
   </PropertyGroup>
 
+  <ItemGroup>
+    <None Remove="Assets\notification-icon-default.ico" />
+    <None Remove="Assets\notification-icon-light.ico" />
+  </ItemGroup>
+
   <ItemGroup>
     <PackageReference Include="Microsoft.Toolkit.Wpf.UI.XamlHost" Version="6.1.2" />
   </ItemGroup>
@@ -22,6 +27,11 @@
     <ProjectReference Include="..\TheXamlGuy.NotificationFlyout.Wpf.UI\TheXamlGuy.NotificationFlyout.Wpf.UI.csproj" />
   </ItemGroup>
 
+  <ItemGroup>
+    <Resource Include="Assets\notification-icon-light.ico" />
+    <Resource Include="Assets\notification-icon-default.ico" />
+  </ItemGroup>
+
   <ItemGroup>
     <PackageReference Update="Microsoft.VCRTForwarders.140" Version="1.0.7" />
   </ItemGroup>
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/XamlHost/XamlHost.cs b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/XamlHost/XamlHost.cs
index 477fe5c..e65755a 100644
--- a/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/XamlHost/XamlHost.cs
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI.Controls/XamlHost/XamlHost.cs
@@ -23,6 +23,7 @@ namespace TheXamlGuy.NotificationFlyout.Wpf.UI.Controls
 
         protected virtual void OnContentLoaded()
         {
+            
         }
 
         protected virtual WindowsXamlHost OnPreparingXamlHost(WindowsXamlHost xamlHost)
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI/Extensions/ImageSourceExtensions.cs b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/Extensions/ImageSourceExtensions.cs
new file mode 100644
index 0000000..1ffdf4c
--- /dev/null
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/Extensions/ImageSourceExtensions.cs
@@ -0,0 +1,31 @@
+using System;
+using System.Drawing;
+using System.Windows;
+using System.Windows.Media;
+using Microsoft.Windows.Sdk;
+
+namespace TheXamlGuy.NotificationFlyout.Wpf.UI.Extensions
+{
+    public static class ImageSourceExtensions
+    {
+        public static Icon ConvertToIcon(this ImageSource imageSource, uint dpi)
+        {
+            if (imageSource == null) return null;
+
+            var uri = new Uri(imageSource.ToString(), UriKind.RelativeOrAbsolute);
+
+            var streamResource = Application.GetResourceStream(uri);
+            if (streamResource == null) throw new ArgumentException(nameof(streamResource));
+
+            return new Icon(streamResource.Stream, new System.Drawing.Size(PInvoke.GetSystemMetricsForDpi((int)SystemMetricFlag.SM_CXICON, dpi), PInvoke.GetSystemMetricsForDpi((int)SystemMetricFlag.SM_CYICON, dpi)));
+        }
+
+        private enum SystemMetricFlag : int
+        {
+            SM_CXICON = 11,
+            SM_CYICON = 12,
+            SM_CXSMICON = 49,
+            SM_CYSMICON = 50
+        }
+    }
+}
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI/NativeMethods.txt b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/NativeMethods.txt
new file mode 100644
index 0000000..5f28270
--- /dev/null
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/NativeMethods.txt
@@ -0,0 +1 @@
+
\ No newline at end of file
diff --git a/src/TheXamlGuy.NotificationFlyout.Wpf.UI/TheXamlGuy.NotificationFlyout.Wpf.UI.csproj b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/TheXamlGuy.NotificationFlyout.Wpf.UI.csproj
index 4d09476..faa27b5 100644
--- a/src/TheXamlGuy.NotificationFlyout.Wpf.UI/TheXamlGuy.NotificationFlyout.Wpf.UI.csproj
+++ b/src/TheXamlGuy.NotificationFlyout.Wpf.UI/TheXamlGuy.NotificationFlyout.Wpf.UI.csproj
@@ -11,6 +11,13 @@
     <Version>1.0.0</Version>
   </PropertyGroup>
 
+  <ItemGroup>
+    <PackageReference Include="Microsoft.Windows.CsWin32" Version="0.1.319-beta">
+      <PrivateAssets>all</PrivateAssets>
+      <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+    </PackageReference>
+  </ItemGroup>
+
   <ItemGroup>
     <ProjectReference Include="..\TheXamlGuy.NotificationFlyout.Shared.UI\TheXamlGuy.NotificationFlyout.Shared.UI.csproj" />
   </ItemGroup>