Use correct types for UFS devices