diff --git a/tools/Legerity.Uno.PageObjectGenerator/Program.cs b/tools/Legerity.Uno.PageObjectGenerator/Program.cs
index 679d1f1..f07db31 100644
--- a/tools/Legerity.Uno.PageObjectGenerator/Program.cs
+++ b/tools/Legerity.Uno.PageObjectGenerator/Program.cs
@@ -8,10 +8,10 @@ namespace Legerity.Uno;
 
 public class Program
 {
-    public static void Main(string[] args)
+    public static async Task Main(string[] args)
     {
         SerilogConfigurator.ConfigureLogging();
-        Parser.Default.ParseArguments<Options>(args)
+        await Parser.Default.ParseArguments<Options>(args)
             .WithNotParsed(errors =>
             {
                 foreach (Error? error in errors)
@@ -30,8 +30,11 @@ public static void Main(string[] args)
                 {
                     Directory.CreateDirectory(options.OutputPath);
                 }
-                
-                await new XamlPageObjectGenerator().GenerateAsync(options.Namespace, options.InputPath, options.OutputPath);
+
+                await new XamlPageObjectGenerator().GenerateAsync(
+                    options.Namespace,
+                    options.InputPath,
+                    options.OutputPath);
 
                 Log.Information("Finished generating Legerity for Uno Platform page objects!");
             });
diff --git a/tools/Legerity.Uno.PageObjectGenerator/Templates/UnoPageObject.template b/tools/Legerity.Uno.PageObjectGenerator/Templates/UnoPageObject.template
index 2f9b4ef..d595c16 100644
--- a/tools/Legerity.Uno.PageObjectGenerator/Templates/UnoPageObject.template
+++ b/tools/Legerity.Uno.PageObjectGenerator/Templates/UnoPageObject.template
@@ -9,7 +9,6 @@ using Legerity.Uno.Extensions;
 using OpenQA.Selenium;
 using OpenQA.Selenium.Appium.Windows;
 using OpenQA.Selenium.Remote;
-using Shouldly;
 using ByExtensions = Legerity.Windows.Extensions.ByExtensions;
 
 namespace {{namespace}}