diff --git a/controllers/message.go b/controllers/message.go
index 2c30e3a..96f2369 100644
--- a/controllers/message.go
+++ b/controllers/message.go
@@ -145,7 +145,7 @@ func (c *ApiController) GetMessageAnswer() {
 	fmt.Printf("Refined Question: [%s]\n", realQuestion)
 	fmt.Printf("Answer: [")
 
-	writer := &RefinedWriter{c.Ctx.ResponseWriter, *NewCleaner(6), []byte{}}
+	writer := &RefinedWriter{*c.Ctx.ResponseWriter, *NewCleaner(6), []byte{}}
 	stringBuilder := &strings.Builder{}
 	err = modelProviderObj.QueryText(realQuestion, writer, stringBuilder)
 	if err != nil {
diff --git a/controllers/message_writer.go b/controllers/message_writer.go
index 215b8f7..49ec292 100644
--- a/controllers/message_writer.go
+++ b/controllers/message_writer.go
@@ -16,17 +16,18 @@ package controllers
 
 import (
 	"fmt"
-	"net/http"
 	"strings"
+
+	"github.com/astaxie/beego/context"
 )
 
 type RefinedWriter struct {
-	http.ResponseWriter
+	context.Response
 	writerCleaner Cleaner
 	buf           []byte
 }
 
-func newRefinedWriter(w http.ResponseWriter) *RefinedWriter {
+func newRefinedWriter(w context.Response) *RefinedWriter {
 	return &RefinedWriter{w, *NewCleaner(6), []byte{}}
 }