diff --git a/t/plugin/traffic-split2.t b/t/plugin/traffic-split2.t
index 05b8047d3f24..4a0da969fa19 100644
--- a/t/plugin/traffic-split2.t
+++ b/t/plugin/traffic-split2.t
@@ -58,7 +58,8 @@ __DATA__
                                     upstream = {
                                         name = "upstream_A",
                                         type = "roundrobin",
-                                        nodes = {["127.0.0.1:1981"] = 1}
+                                        nodes = {["127.0.0.1:1981"] = 1},
+                                        scheme = "http"
                                     },
                                     weight = 1,
                                 }
@@ -68,7 +69,8 @@ __DATA__
                 },
                 upstream = {
                     type = "roundrobin",
-                    nodes = {["127.0.0.1:1980"] = 1}
+                    nodes = {["127.0.0.1:1980"] = 1},
+                    scheme = "http"
                 }
             }
             local code, body = t('/apisix/admin/routes/1',
@@ -278,7 +280,7 @@ GET /uri?name=jack
 host: 127.0.0.1
 --- response_body
 uri: /uri
-host: localhost
+host: localhost:1981
 x-real-ip: 127.0.0.1