1 files changed, 18 insertions, 5 deletions
diff --git a/src/lua.h b/src/lua.h
index 6b146f6..625a0c1 100644
--- a/src/lua.h
+++ b/src/lua.h
@@ -1,8 +1,21 @@
-#include <lua5.4/lua.h>
-#include <lua5.4/lualib.h>
-#include <lua5.4/lauxlib.h>
+#include <lua.h>
+#include <lualib.h>
+#include <lauxlib.h>
 
-#define lua_objlen(L,i) lua_rawlen(L,(i))
-#define luaL_register(L, M, F) luaL_newlib(L, F);
+
+#if LUA_VERSION_NUM == 504
+    #define lreg(N, FN)\
+        lua_pushstring(L, N);\
+        luaL_register(L, NULL, FN);\
+        lua_settable(L, -3);
+
+    #define lua_objlen(L,i) lua_rawlen(L,(i))
+    #define luaL_register(L, M, F) luaL_newlib(L, F);
+#else
+    #define lreg(N, FN)\
+        lua_newtable(L);\
+        luaL_register(L, NULL, FN);\
+        lua_setfield(L, 2, N);
+#endif