diff --git a/app/Controllers/UpgradeController.php b/app/Controllers/UpgradeController.php
index fad1161..7b58f66 100644
--- a/app/Controllers/UpgradeController.php
+++ b/app/Controllers/UpgradeController.php
@@ -93,7 +93,11 @@ class UpgradeController extends Controller
         }
 
         foreach ($currentFiles as $extraneous) {
-            unlink($extraneous);
+            if (is_dir($extraneous)) {
+                removeDirectory($extraneous);
+            } else {
+                unlink($extraneous);
+            }
         }
 
         $updateZip->close();
@@ -105,15 +109,15 @@ class UpgradeController extends Controller
     }
 
     /**
-     * @param Request  $request
-     * @param Response $response
+     * @param  Request  $request
+     * @param  Response  $response
      *
      * @return Response
      */
     public function checkForUpdates(Request $request, Response $response): Response
     {
         $jsonResponse = [
-            'status'  => null,
+            'status' => null,
             'message' => null,
             'upgrade' => false,
         ];
diff --git a/composer.json b/composer.json
index 0ee52ee..a6a0eb6 100644
--- a/composer.json
+++ b/composer.json
@@ -1,6 +1,6 @@
 {
   "name": "sergix44/xbackbone",
-  "version": "3.1.RC3",
+  "version": "3.1.RC4",
   "description": "A lightweight ShareX PHP backend",
   "type": "project",
   "require": {