1 files changed, 3 insertions, 5 deletions
diff --git a/askama_shared/src/generator.rs b/askama_shared/src/generator.rs
index 368db9f..738b878 100644
--- a/askama_shared/src/generator.rs
+++ b/askama_shared/src/generator.rs
@@ -227,12 +227,10 @@ impl<'a, S: std::hash::BuildHasher> Generator<'a, S> {
 
     // Implement Axum's `IntoResponse`.
     fn impl_axum_into_response(&mut self, buf: &mut Buffer) -> Result<(), CompileError> {
-        self.write_header(buf, "::axum::response::IntoResponse", None)?;
+        self.write_header(buf, "::askama_axum::IntoResponse", None)?;
         buf.writeln(
-            "type Body = ::axum::body::Full<::axum::body::Bytes>;\n\
-            type BodyError = std::convert::Infallible;\n\
-            fn into_response(self)\
-             -> ::axum::http::Response<Self::Body> {",
+            "fn into_response(self)\
+             -> ::askama_axum::Response<::askama_axum::BoxBody> {",
         )?;
         let ext = self.input.extension().unwrap_or("txt");
         buf.writeln(&format!("::askama_axum::into_response(&self, {:?})", ext))?;