diff --git a/src/lib.rs b/src/lib.rs
index a21728a..2ca2cc7 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -193,7 +193,7 @@ impl Client {
 
     /// Send a chat message to a model.
     /// ```no_run
-    /// use aleph_alpha_client::{Client, How, TaskChat, Error};
+    /// use aleph_alpha_client::{Client, How, TaskChat, Error, Role};
     ///
     /// async fn chat() -> Result<(), Error> {
     ///     // Authenticate against API. Fetches token.