diff --git a/src/ESPAsyncWebServer.h b/src/ESPAsyncWebServer.h
index 52b2d5d..390af1c 100644
--- a/src/ESPAsyncWebServer.h
+++ b/src/ESPAsyncWebServer.h
@@ -279,7 +279,7 @@ class AsyncWebRewrite {
         _toUrl = _toUrl.substring(0, index);
       }
     }
-    AsyncWebRewrite& setFilter(ArRequestFilterFunction fn) { _filter = fn; }
+    AsyncWebRewrite& setFilter(ArRequestFilterFunction fn) { _filter = fn; return *this; }
     bool filter(AsyncWebServerRequest *request){ return _filter == NULL || _filter(request); }
     String from(void) { return _from; }
     String toUrl(void) { return _toUrl; }
@@ -296,7 +296,7 @@ class AsyncWebHandler {
   public:
     AsyncWebHandler* next;
     AsyncWebHandler(): next(NULL){}
-    AsyncWebHandler& setFilter(ArRequestFilterFunction fn) { _filter = fn; }
+    AsyncWebHandler& setFilter(ArRequestFilterFunction fn) { _filter = fn; return *this; }
     bool filter(AsyncWebServerRequest *request){ return _filter == NULL || _filter(request); }
     virtual ~AsyncWebHandler(){}
     virtual bool canHandle(AsyncWebServerRequest *request){ return false; }
@@ -330,6 +330,7 @@ class AsyncWebServerResponse {
   public:
     AsyncWebServerResponse();
     virtual ~AsyncWebServerResponse();
+    virtual void setCode(int code);
     virtual void setContentLength(size_t len);
     virtual void setContentType(String type);
     virtual void addHeader(String name, String value);
diff --git a/src/WebResponses.cpp b/src/WebResponses.cpp
index 791ab98..1d9355e 100644
--- a/src/WebResponses.cpp
+++ b/src/WebResponses.cpp
@@ -92,6 +92,11 @@ AsyncWebServerResponse::~AsyncWebServerResponse(){
   }
 }
 
+void AsyncWebServerResponse::setCode(int code){
+  if(_state == RESPONSE_SETUP)
+    _code = code;
+}
+
 void AsyncWebServerResponse::setContentLength(size_t len){
   if(_state == RESPONSE_SETUP)
     _contentLength = len;