diff --git a/src/FileUploadControl.php b/src/FileUploadControl.php
index 6cf2669..6b242cd 100644
--- a/src/FileUploadControl.php
+++ b/src/FileUploadControl.php
@@ -67,7 +67,7 @@ class FileUploadControl extends UploadControl
 	/** @var UploadController */
 	private $controller;
 
-	/** @var string|null */
+	/** @var class-string<IUploadModel>|null */
 	private $uploadModel;
 
 	/**
@@ -314,6 +314,9 @@ public function getUploadModel(): IUploadModel
 		}
 	}
 
+	/**
+	 * @param class-string<IUploadModel> $uploadModel
+	 */
 	public function setUploadModel(string $uploadModel): self
 	{
 		$this->uploadModel = $uploadModel;