diff --git a/viz-core/src/handler/after.rs b/viz-core/src/handler/after.rs
index a49f299f..95e95204 100644
--- a/viz-core/src/handler/after.rs
+++ b/viz-core/src/handler/after.rs
@@ -21,7 +21,7 @@ where
     I: Send + 'static,
     H: Handler<I, Output = Result<O>> + Clone,
     O: Send + 'static,
-    F: Handler<Result<O>, Output = Result<O>> + Clone,
+    F: Handler<H::Output, Output = H::Output> + Clone,
 {
     type Output = F::Output;
 
diff --git a/viz-test/tests/body.rs b/viz-test/tests/body.rs
index 2386f297..60f3aff6 100644
--- a/viz-test/tests/body.rs
+++ b/viz-test/tests/body.rs
@@ -159,7 +159,7 @@ async fn outgoing_body() -> Result<()> {
     assert!(full_some.frame().await.is_none());
 
     let mut boxed: Body = UnsyncBoxBody::new(Full::new(Bytes::new()).map_err(Into::into)).into();
-    assert_eq!(boxed.is_end_stream(), false);
+    assert!(!boxed.is_end_stream());
     // boxed stream uses default size
     let size_hint = boxed.size_hint();
     assert_eq!(size_hint.lower(), 0);