diff --git a/pymobiledevice3/restore/tss.py b/pymobiledevice3/restore/tss.py
index 105493bd8..4868fb611 100644
--- a/pymobiledevice3/restore/tss.py
+++ b/pymobiledevice3/restore/tss.py
@@ -351,14 +351,9 @@ def add_ap_img4_tags(self, parameters):
                     k = 'SepNonce'
                 self._request[k] = v
 
-        uid_mode = parameters.get('UID_MODE')
-        requires_uid_mode = parameters.get('RequiresUIDMode')
-        if uid_mode is not None:
-            self._request['UID_MODE'] = uid_mode
-        elif requires_uid_mode is not None:
-            # The logic here is missing why this value is expected to be 'false'
-            self._request['UID_MODE'] = False
+        uid_mode = parameters.get('UID_MODE', False)
 
+        self._request['UID_MODE'] = uid_mode
         self._request['@ApImg4Ticket'] = True
         self._request['@BBTicket'] = True