1 files changed, 2 insertions, 5 deletions
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index 8f2866e..4a6eb89 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -234,11 +234,8 @@ impl<'a> Generator<'a> {
              -> ::std::result::Result<Self::Item, Self::Error> {",
         );
 
-        let ext = match self.input.path.extension() {
-            Some(s) => s.to_str().unwrap(),
-            None => "txt",
-        };
-        buf.writeln(&format!("::askama::actix_web::respond(&self, {:?})", ext));
+        buf.writeln("use ::askama::actix_web::TemplateIntoResponse;");
+        buf.writeln("self.into_response()");
 
         buf.writeln("}");
         buf.writeln("}");