diff --git a/src/Database/Model.php b/src/Database/Model.php
index 3ff76fe..7f8bed6 100644
--- a/src/Database/Model.php
+++ b/src/Database/Model.php
@@ -97,9 +97,10 @@
 		 *
 		 * @param string $key Target column to update
 		 * @param mixed $value New value of target column
+		 * @param return $return (optional) Return this instead of $value when set
 		 * @return mixed The value that was sent to $value
 		 */
-		public function set(string $key, mixed $value): mixed {
+		public function set(string $key, mixed $value, mixed $return = null): mixed {
 			$this->_row[$key] = $value;
 
 			$this->db
@@ -107,6 +108,6 @@
 				->where($this->where)
 				->update([$key => $value]);
 
-			return $value;
+			return $return ?? $value;
 		}
 	}