diff --git a/IntegrationTest/lua/mysqloo/tests/prepared_query_tests.lua b/IntegrationTest/lua/mysqloo/tests/prepared_query_tests.lua
index 8871e8e..590318f 100644
--- a/IntegrationTest/lua/mysqloo/tests/prepared_query_tests.lua
+++ b/IntegrationTest/lua/mysqloo/tests/prepared_query_tests.lua
@@ -181,9 +181,10 @@ end)
 TestFramework:RegisterTest("[Prepared Query] should return correct error", function(test)
 	local db = TestFramework:ConnectToDatabase()
 	local qu = db:prepare("SEsdg")
-	function qu:onError(err)
+	function qu:onError(err, sql)
 		test:shouldBeEqual(qu:error(), err)
 		test:shouldBeGreaterThan(#qu:error(), 0)
+		test:shouldBeEqual(sql, "SEsdg")
 		test:Complete()
 	end
 	qu:start()
diff --git a/IntegrationTest/lua/mysqloo/tests/query_tests.lua b/IntegrationTest/lua/mysqloo/tests/query_tests.lua
index d2a9502..ae86fb2 100644
--- a/IntegrationTest/lua/mysqloo/tests/query_tests.lua
+++ b/IntegrationTest/lua/mysqloo/tests/query_tests.lua
@@ -121,8 +121,9 @@ end)
 TestFramework:RegisterTest("[Query] should return correct error", function(test)
 	local db = TestFramework:ConnectToDatabase()
 	local qu = db:query("SEsdg")
-	function qu:onError(err)
+	function qu:onError(err, sql)
 		test:shouldBeEqual(qu:error(), err)
+		test:shouldBeEqual(sql, "SEsdg")
 		test:shouldBeGreaterThan(#qu:error(), 0)
 		test:Complete()
 	end