diff --git a/src/lib.rs b/src/lib.rs
index 417bad8..f340ec2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -54,6 +54,8 @@
     InvalidSuccessArgsVariant,
     #[error("Invalid notification count")]
     InvalidNotificationCount,
+    #[error("Invalid Partition Info Get Regs response")]
+    InvalidPartitionInfoGetRegsResponse,
 }
 
 impl From<Error> for FfaError {
@@ -73,7 +75,8 @@
             | Error::UnrecognisedWarmBootType(_)
             | Error::InvalidPartitionInfoGetFlag(_)
             | Error::InvalidSuccessArgsVariant
-            | Error::InvalidNotificationCount => Self::InvalidParameters,
+            | Error::InvalidNotificationCount
+            | Error::InvalidPartitionInfoGetRegsResponse => Self::InvalidParameters,
         }
     }
 }
@@ -212,6 +215,7 @@
 /// * `FFA_ID_GET` - [`SuccessArgsIdGet`]
 /// * `FFA_SPM_ID_GET` - [`SuccessArgsSpmIdGet`]
 /// * `FFA_PARTITION_INFO_GET` - [`partition_info::SuccessArgsPartitionInfoGet`]
+/// * `FFA_PARTITION_INFO_GET_REGS` - [`partition_info::SuccessArgsPartitionInfoGetRegs`]
 /// * `FFA_NOTIFICATION_GET` - [`SuccessArgsNotificationGet`]
 /// * `FFA_NOTIFICATION_INFO_GET_32` - [`SuccessArgsNotificationInfoGet32`]
 /// * `FFA_NOTIFICATION_INFO_GET_64` - [`SuccessArgsNotificationInfoGet64`]