diff --git a/whitechapel_pro/device.te b/whitechapel_pro/device.te
index ae74fea2..24bb1e8a 100644
--- a/whitechapel_pro/device.te
+++ b/whitechapel_pro/device.te
@@ -1,3 +1,4 @@
+# device.te
 type modem_block_device, dev_type;
 type custom_ab_block_device, dev_type;
 type persist_block_device, dev_type;
@@ -20,3 +21,6 @@ type fips_block_device, dev_type;
 # SecureElement SPI device
 type st54spi_device, dev_type;
 type st33spi_device, dev_type;
+
+# Storage firmware upgrade
+type ufs_internal_block_device, dev_type;