1 files changed, 15 insertions, 12 deletions
diff --git a/Sluift/Lua/Check.h b/Sluift/Lua/Check.h
index a569826..61658a6 100644
--- a/Sluift/Lua/Check.h
+++ b/Sluift/Lua/Check.h
@@ -1,6 +1,6 @@
 /*
- * Copyright (c) 2013 Remko Tronçon
- * Licensed under the GNU General Public License.
+ * Copyright (c) 2013-2014 Isode Limited.
+ * All rights reserved.
  * See the COPYING file for more information.
  */
 
@@ -8,19 +8,22 @@
 
 #include <string>
 
+#include <Swiften/Base/ByteArray.h>
+
 struct lua_State;
 
 namespace Swift {
-	namespace Lua {
-		void checkType(lua_State* L, int arg, int type);
-		int checkIntNumber(lua_State* L, int arg);
-		std::string checkString(lua_State* L, int arg);
+    namespace Lua {
+        void checkType(lua_State* L, int arg, int type);
+        int checkIntNumber(lua_State* L, int arg);
+        std::string checkString(lua_State* L, int arg);
+        ByteArray checkByteArray(lua_State* L, int arg);
 
-		void* checkUserDataRaw(lua_State* L, int arg, const char* tableName);
+        void* checkUserDataRaw(lua_State* L, int arg);
 
-		template<typename T>
-		T** checkUserData(lua_State* L, int arg, const char* tableName) {
-			return reinterpret_cast<T**>(checkUserDataRaw(L, arg, tableName));
-		}
-	}
+        template<typename T>
+        T** checkUserData(lua_State* L, int arg) {
+            return reinterpret_cast<T**>(checkUserDataRaw(L, arg));
+        }
+    }
 }