diff --git a/Directory.Packages.props b/Directory.Packages.props
index f0776fd..dd9048f 100644
--- a/Directory.Packages.props
+++ b/Directory.Packages.props
@@ -4,7 +4,7 @@
   </PropertyGroup>
   <ItemGroup>
     <PackageVersion Include="Avalonia" Version="11.0.10" />
-    <PackageVersion Include="Avalonia.Controls.TreeDataGrid" Version="11.0.2" />
+    <PackageVersion Include="Avalonia.Controls.TreeDataGrid" Version="11.0.10" />
     <PackageVersion Include="Avalonia.Desktop" Version="11.0.10" />
     <PackageVersion Include="Avalonia.Themes.Fluent" Version="11.0.10" />
     <PackageVersion Include="Avalonia.Fonts.Inter" Version="11.0.10" />
diff --git a/samples/SampleApp/App.axaml b/samples/SampleApp/App.axaml
index 4954d99..4277ec8 100644
--- a/samples/SampleApp/App.axaml
+++ b/samples/SampleApp/App.axaml
@@ -6,7 +6,6 @@
 
     <Application.Styles>
         <FluentTheme />
-        <!-- <StyleInclude Source="avares://Avalonia.Controls.TreeDataGrid/Themes/Fluent.axaml"/> -->
         <StyleInclude Source="avares://DynamicTreeDataGrid/Themes/Generic.axaml"/>
 
     </Application.Styles>
diff --git a/src/DynamicTreeDataGrid/Themes/Generic.axaml b/src/DynamicTreeDataGrid/Themes/Generic.axaml
index ec27824..f97b8e5 100644
--- a/src/DynamicTreeDataGrid/Themes/Generic.axaml
+++ b/src/DynamicTreeDataGrid/Themes/Generic.axaml
@@ -11,9 +11,7 @@
 		</Border>
 	</Design.PreviewWith>
 
-	<!-- TODO: Use the Generic style once it's exposed. -->
-	<!-- <StyleInclude Source="avares://Avalonia.Controls.TreeDataGrid/Themes/Generic.axaml"/> -->
-	<StyleInclude Source="avares://Avalonia.Controls.TreeDataGrid/Themes/Fluent.axaml" />
+	<StyleInclude Source="avares://Avalonia.Controls.TreeDataGrid/Themes/Generic.axaml"/>
 
 	<Style Selector="FlyoutPresenter.NoPadding">
 		<Setter Property="Padding" Value="0" />