diff --git a/src/blocking.rs b/src/blocking.rs
index b543dcd..5d8d17d 100644
--- a/src/blocking.rs
+++ b/src/blocking.rs
@@ -182,6 +182,8 @@ impl<S: Read + Write> Client<S> {
         match self.handle(Request::Extension(extension))? {
             Response::Success => Ok(None),
             Response::ExtensionResponse(response) => Ok(Some(response)),
+            Response::Failure => Err(AgentError::Failure),
+            Response::ExtensionFailure => Err(AgentError::ExtensionFailure),
             _ => Err(ProtoError::UnexpectedResponse.into()),
         }
     }
diff --git a/src/client.rs b/src/client.rs
index 1618dee..22e0048 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -186,6 +186,8 @@ where
         match self.handle(Request::Extension(extension)).await? {
             Response::Success => Ok(None),
             Response::ExtensionResponse(response) => Ok(Some(response)),
+            Response::Failure => Err(AgentError::Failure),
+            Response::ExtensionFailure => Err(AgentError::ExtensionFailure),
             _ => Err(ProtoError::UnexpectedResponse.into()),
         }
     }