diff --git a/stream/config b/stream/config
index 08c4a810..bfd39c1b 100644
--- a/stream/config
+++ b/stream/config
@@ -1,7 +1,13 @@
 ngx_module_type=STREAM
 ngx_module_name=ngx_stream_lua_kong_module
 ngx_module_srcs="$ngx_addon_dir/src/ngx_stream_lua_kong_module.c"
-ngx_module_incs="$ngx_addon_dir/src"
+
+if [ -z "$NGX_STREAM_LUA_LOC" ]; then
+    ngx_stream_lua_loc="../ngx_stream_lua-*"
+else
+    ngx_stream_lua_loc="$NGX_STREAM_LUA_LOC"
+fi
+ngx_module_incs="$ngx_addon_dir/src $ngx_stream_lua_loc/src"
 
 . auto/module
 
diff --git a/stream/src/ngx_stream_lua_kong_module.c b/stream/src/ngx_stream_lua_kong_module.c
index 3ebf63b6..7b1f48f5 100644
--- a/stream/src/ngx_stream_lua_kong_module.c
+++ b/stream/src/ngx_stream_lua_kong_module.c
@@ -19,6 +19,7 @@
 #include <ngx_core.h>
 #include <ngx_stream.h>
 #include <ngx_stream_lua_api.h>
+#include <ngx_stream_lua_socket_tcp.h>
 #include "ngx_stream_lua_kong_module.h"