1 files changed, 6 insertions, 4 deletions

diff --git a/src/luasocket.c b/src/luasocket.c
index 94ea05b..434b5b7 100644
--- a/src/luasocket.c
+++ b/src/luasocket.c
@@ -17,15 +17,17 @@
 /*=========================================================================*\
 * Standard include files
 \*=========================================================================*/
-#include <lua.h>
-#include <lauxlib.h>
+#include "lua.h"
+#include "lauxlib.h"
+
+#if !defined(LUA_VERSION_NUM) || (LUA_VERSION_NUM < 501)
 #include "compat-5.1.h"
-#include "luasocket.h"
+#endif
 
 /*=========================================================================*\
 * LuaSocket includes
 \*=========================================================================*/
-
+#include "luasocket.h"
 #include "auxiliar.h"
 #include "except.h"
 #include "timeout.h"