diff --git a/context/output.go b/context/output.go
index d132e392..0a194753 100644
--- a/context/output.go
+++ b/context/output.go
@@ -66,7 +66,6 @@ func (output *BeegoOutput) Body(content []byte) {
 			if val == "gzip" {
 				output.Header("Content-Encoding", "gzip")
 				outputWriter, _ = gzip.NewWriterLevel(output.Context.ResponseWriter, gzip.BestSpeed)
-
 				break
 			} else if val == "deflate" {
 				output.Header("Content-Encoding", "deflate")
@@ -86,11 +85,8 @@ func (output *BeegoOutput) Body(content []byte) {
 	}
 
 	outputWriter.Write(content)
-	switch outputWriter.(type) {
-	case *gzip.Writer:
-		outputWriter.(*gzip.Writer).Close()
-	case *flate.Writer:
-		outputWriter.(*flate.Writer).Close()
+	if c, ok := outputWriter.(io.Closer); ok {
+		c.Close()
 	}
 }
 
diff --git a/router.go b/router.go
index 1e8144bf..5c6b7bf9 100644
--- a/router.go
+++ b/router.go
@@ -897,10 +897,6 @@ type responseWriter struct {
 	status  int
 }
 
-// Header returns the header map that will be sent by WriteHeader.
-func (w *responseWriter) Header() http.Header {
-	return w.ResponseWriter.Header()
-}
 
 // Write writes the data to the connection as part of an HTTP reply,
 // and sets `started` to true.