diff --git a/cpanel/gniza/index.live.cgi b/cpanel/gniza/index.live.cgi
index 05aedd8..a25b89c 100644
--- a/cpanel/gniza/index.live.cgi
+++ b/cpanel/gniza/index.live.cgi
@@ -14,9 +14,11 @@ BEGIN {
     unshift @INC, "$base/lib";
 }
 
+use Cpanel::LiveAPI ();
 use Cpanel::AdminBin::Call ();
 use GnizaCPanel::UI;
 
+my $cpanel = Cpanel::LiveAPI->new();
 print "Content-Type: text/html\r\n\r\n";
 
 # Get allowed remotes via AdminBin
@@ -28,6 +30,7 @@ print GnizaCPanel::UI::page_header('gniza Restore');
 if (!@remotes) {
     print qq{<div class="alert alert-info mb-4">No backup remotes are available for restore. Please contact your server administrator.</div>\n};
     print GnizaCPanel::UI::page_footer();
+    $cpanel->end();
     exit;
 }
 
@@ -103,3 +106,4 @@ for my $cat (@categories) {
 print qq{</div>\n};
 
 print GnizaCPanel::UI::page_footer();
+$cpanel->end();
diff --git a/cpanel/gniza/restore.live.cgi b/cpanel/gniza/restore.live.cgi
index a6c2388..46462f7 100644
--- a/cpanel/gniza/restore.live.cgi
+++ b/cpanel/gniza/restore.live.cgi
@@ -14,10 +14,13 @@ BEGIN {
     unshift @INC, "$base/lib";
 }
 
+use Cpanel::LiveAPI ();
 use Cpanel::AdminBin::Call ();
 use Cpanel::Form           ();
 use GnizaCPanel::UI;
 
+my $cpanel = Cpanel::LiveAPI->new();
+END { $cpanel->end() if $cpanel }
 my $form   = Cpanel::Form::parseform();
 my $method = $ENV{'REQUEST_METHOD'} // 'GET';
 my $step   = $form->{'step'} // '1';