diff --git a/src/lib.rs b/src/lib.rs
index 5d1fc2e..1a0dfc9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -8,9 +8,9 @@
 //!
 //! ## Fetching an Ethereum Block
 //!
-//! ```no_run
+//! ```rust,no_run
 //! # use firehose_client::{Chain, FirehoseClient};
-//! # use vee::EthBlock as Block;
+//! # use vee::protos::EthBlock as Block;
 //! # #[tokio::main]
 //! # async fn main() -> Result<(), vee::ProtosError> {
 //! let mut client = FirehoseClient::new(Chain::Ethereum);
@@ -29,7 +29,7 @@
 //!
 //! ## Streaming Ethereum Blocks
 //!
-//! ```no_run
+//! ```rust,no_run
 //! # use firehose_client::{Chain, FirehoseClient};
 //! # use futures::StreamExt;
 //! # #[tokio::main]