diff --git a/libraries/pipes/implementations/rust/src/context_loader.rs b/libraries/pipes/implementations/rust/src/context_loader.rs
index 544c0eb..e319fcb 100644
--- a/libraries/pipes/implementations/rust/src/context_loader.rs
+++ b/libraries/pipes/implementations/rust/src/context_loader.rs
@@ -5,6 +5,7 @@ use thiserror::Error;
 
 use crate::PipesContextData;
 
+/// Load context data injected by the orchestration process.
 pub trait LoadContext {
     fn load_context(
         &self,
diff --git a/libraries/pipes/implementations/rust/src/lib.rs b/libraries/pipes/implementations/rust/src/lib.rs
index be1d20f..0b5ccc8 100644
--- a/libraries/pipes/implementations/rust/src/lib.rs
+++ b/libraries/pipes/implementations/rust/src/lib.rs
@@ -12,18 +12,19 @@ use serde_json::Map;
 use serde_json::Value;
 use thiserror::Error;
 
-use crate::context_loader::DefaultLoader as PipesDefaultContextLoader;
+use crate::context_loader::{DefaultLoader as PipesDefaultContextLoader, PayloadErrorKind};
+use crate::params_loader::{EnvVarLoader as PipesEnvVarParamsLoader, ParamsError};
+use crate::types::{Method, PipesContextData, PipesMessage};
+use crate::writer::message_writer::{
+    get_opened_payload, DefaultWriter as PipesDefaultMessageWriter,
+};
+use crate::writer::message_writer_channel::MessageWriteError;
+
 pub use crate::context_loader::LoadContext;
-use crate::context_loader::PayloadErrorKind;
-use crate::params_loader::EnvVarLoader as PipesEnvVarParamsLoader;
 pub use crate::params_loader::LoadParams;
-use crate::params_loader::ParamsError;
 pub use crate::types::PipesMetadataValue;
-use crate::types::{Method, PipesContextData, PipesMessage};
-use crate::writer::message_writer::get_opened_payload;
-use crate::writer::message_writer::DefaultWriter as PipesDefaultMessageWriter;
 pub use crate::writer::message_writer::{DefaultWriter, MessageWriter};
-use crate::writer::message_writer_channel::{MessageWriteError, MessageWriterChannel};
+pub use crate::writer::message_writer_channel::MessageWriterChannel;
 
 #[derive(Serialize)]
 #[serde(rename_all = "UPPERCASE")]
diff --git a/libraries/pipes/implementations/rust/src/params_loader.rs b/libraries/pipes/implementations/rust/src/params_loader.rs
index 5be23ba..92a3aaf 100644
--- a/libraries/pipes/implementations/rust/src/params_loader.rs
+++ b/libraries/pipes/implementations/rust/src/params_loader.rs
@@ -10,8 +10,8 @@ const DAGSTER_PIPES_CONTEXT_ENV_VAR: &str = "DAGSTER_PIPES_CONTEXT";
 const DAGSTER_PIPES_MESSAGES_ENV_VAR: &str = "DAGSTER_PIPES_MESSAGES";
 
 /// Load params passed from the orchestration process by the context injector and
-/// message reader. These params are used to respectively bootstrap
-/// [`LoadContext`](crate::LoadContext) and [`PipesMessageWriter`].
+/// message reader. These params are used to respectively bootstrap implementations of
+/// [`LoadContext`](crate::LoadContext) and [`PipesMessageWriter`](crate::MessageWriter).
 pub trait LoadParams {
     /// Whether or not this process has been provided with provided with information
     /// to create a `PipesContext` or should instead return a mock.
diff --git a/libraries/pipes/implementations/rust/src/writer/message_writer.rs b/libraries/pipes/implementations/rust/src/writer/message_writer.rs
index d1094e8..ee9fa06 100644
--- a/libraries/pipes/implementations/rust/src/writer/message_writer.rs
+++ b/libraries/pipes/implementations/rust/src/writer/message_writer.rs
@@ -11,7 +11,7 @@ mod private {
     pub struct Token; // To seal certain trait methods
 }
 
-/// Write messages back to Dagster, via its associated [`Self::Channel`].
+/// Write messages back to Dagster, via its associated [`MessageWriterChannel`](crate::MessageWriterChannel).
 pub trait MessageWriter {
     type Channel: MessageWriterChannel;