diff --git a/app/Controllers/UploadController.php b/app/Controllers/UploadController.php
index d25ce9c..34d0bc2 100644
--- a/app/Controllers/UploadController.php
+++ b/app/Controllers/UploadController.php
@@ -118,7 +118,6 @@ public function uploadEndpoint(Request $request, Response $response): Response
     /**
      * @param  Request  $request
      * @param  Response  $response
-     * @param $code
      * @return UploadedFileInterface
      * @throws ValidationException
      */
@@ -179,11 +178,12 @@ protected function validateUser(Request $request, Response $response, UploadedFi
      * @param  Response  $response
      * @param  UploadedFileInterface  $file
      * @param $user
+     * @param $code
      * @return Response
      * @throws \League\Flysystem\FileExistsException
      * @throws \League\Flysystem\FileNotFoundException
      */
-    protected function saveMedia(Response $response, UploadedFileInterface $file, $user, $code)
+    protected function saveMedia(Response $response, UploadedFileInterface $file, $user, $code = null)
     {
         if ($code === null) {
             do {