From a57b5855713e1a8355483a8927ec697d4a7bf60d Mon Sep 17 00:00:00 2001
From: Isaac Daly <isaac.daly@outlook.com>
Date: Fri, 11 Oct 2024 07:45:49 +1100
Subject: [PATCH] Fixed tests

---
 src/Whim.Yaml/YamlBarPluginLoader.cs | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/Whim.Yaml/YamlBarPluginLoader.cs b/src/Whim.Yaml/YamlBarPluginLoader.cs
index d9ba79a97..ef447d641 100644
--- a/src/Whim.Yaml/YamlBarPluginLoader.cs
+++ b/src/Whim.Yaml/YamlBarPluginLoader.cs
@@ -85,7 +85,7 @@ private static List<BarComponent> GetBarComponents(IContext ctx, Schema.DefsRequ
 					components.Add(CreateWorkspaceBarWidget(ctx, widget));
 					return null;
 				},
-				(in Schema.AWidgetToDisplayTheDirectionOfTheTreeLayoutEngine widget) =>
+				(in Schema.DefsRequiredType2 widget) =>
 				{
 					components.Add(CreateTreeLayoutEngineBarWidget(ctx, widget));
 					return null;
@@ -136,10 +136,7 @@ private static BarComponent CreateWorkspaceBarWidget(IContext ctx, Schema.AWidge
 		return WorkspaceWidget.CreateComponent();
 	}
 
-	private static BarComponent CreateTreeLayoutEngineBarWidget(
-		IContext ctx,
-		Schema.AWidgetToDisplayTheDirectionOfTheTreeLayoutEngine widget
-	)
+	private static BarComponent CreateTreeLayoutEngineBarWidget(IContext ctx, Schema.DefsRequiredType2 widget)
 	{
 		if (
 			ctx.PluginManager.LoadedPlugins.FirstOrDefault(p => p.Name == "whim.tree_layout")