1 files changed, 2 insertions, 6 deletions
diff --git a/PublishMedia.cs b/PublishMedia.cs
index 523b00f..003b48a 100644
--- a/PublishMedia.cs
+++ b/PublishMedia.cs
@@ -1,14 +1,10 @@
-using HyperBooru.ApiClient;
-using System.Management.Automation;
+using System.Management.Automation;
 
 namespace HyperBooru.PowerShell;
 
 [Alias("pbhbm")]
 [Cmdlet(VerbsData.Publish, "HyperBooruMedia")]
-public class PublishMediaCmdlet : PSCmdlet {
-    [Parameter(Position = 0, Mandatory = true)]
-    public HBSession Session { get; set; }
-
+public class PublishMediaCmdlet : SessionCmdlet {
     [Parameter(Position = 1, Mandatory = true, ValueFromPipeline = true)]
     public string Path { get; set; }