*update

1 files changed, 5 insertions, 3 deletions

diff --git a/src/lua/net/luaopen_net.cpp b/src/lua/net/luaopen_net.cpp
index 7ec9cc7..0fce20f 100644
--- a/src/lua/net/luaopen_net.cpp
+++ b/src/lua/net/luaopen_net.cpp
@@ -2,13 +2,13 @@
 #include "libjin/jin.h"
 #include "../luaopen_types.h"
 #include "Buffer.h"
-#include "Socket.h"
 
 namespace jin
 {
 namespace lua
 {
     using namespace jin::lua::net;
+	using namespace jin::net;
 
     static int l_initNetwork(lua_State* L)
     {
@@ -47,8 +47,9 @@ namespace lua
             }
         }
         Socket* socket = new Socket(info);
+		Ref<Socket>* ref = new Ref<Socket>(socket);
         Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_SOCKET, sizeof(Proxy));
-        proxy->bind(socket, JIN_NETWORK_SOCKET);
+        proxy->bind(ref, JIN_NETWORK_SOCKET);
         return 1;
     }
 
@@ -58,7 +59,8 @@ namespace lua
         int size = luax_checkinteger(L, 1);
         Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_BUFFER, sizeof(Proxy));
         net::Buffer* buffer = new net::Buffer(size);
-        proxy->bind(buffer, JIN_NETWORK_BUFFER);
+		Ref<Buffer>* ref = new Ref<Buffer>(buffer);
+        proxy->bind(ref, JIN_NETWORK_BUFFER);
         return 1;
     }