1 files changed, 2 insertions, 6 deletions
diff --git a/SetMediaTag.cs b/SetMediaTag.cs
index 8931991..4ff016f 100644
--- a/SetMediaTag.cs
+++ b/SetMediaTag.cs
@@ -1,14 +1,10 @@
-using HyperBooru.ApiClient;
-using System.Management.Automation;
+using System.Management.Automation;
 
 namespace HyperBooru.PowerShell;
 
 [Alias("shbmt")]
 [Cmdlet(VerbsCommon.Set, "HyperBooruMediaTag")]
-public class SetMediaTagCmdlet : PSCmdlet {
-    [Parameter(Position = 0, Mandatory = true)]
-    public required HBSession Session { get; set; }
-
+public class SetMediaTagCmdlet : SessionCmdlet {
     [Parameter(Position = 1, Mandatory = true)]
     public Guid MediaId { get; set; }