diff --git a/classes/query_builder.php b/classes/query_builder.php
index 55b9997..d397a8f 100644
--- a/classes/query_builder.php
+++ b/classes/query_builder.php
@@ -1039,9 +1039,11 @@ class Query_Builder implements iQuery_Builder {
 		// Can't use normal set, because it doesn't handle multidimensional arrays
 		foreach($data as $key => $arr)
 		{
+			$this->set_array_keys[$key] = array();
+		
 			foreach($arr as $k => $v)
 			{
-				$this->set_array_keys[$key][] = $k;
+				array_push($this->set_array_keys[$key], $k);
 				$this->values[] = $v;
 			}
 			
diff --git a/tests/core/db_qb_test.php b/tests/core/db_qb_test.php
index fc0d683..95e94ed 100644
--- a/tests/core/db_qb_test.php
+++ b/tests/core/db_qb_test.php
@@ -465,17 +465,17 @@ abstract class QBTest extends UnitTestCase {
 		
 		$insert_array = array(
 			array(
-				'id' => 6,
+				'id' => 11,
 				'key' => 2,
 				'val' => 3
 			),
 			array(
-				'id' => 5,
+				'id' => 12,
 				'key' => 6,
 				'val' => 7
 			),
 			array(
-				'id' => 8,
+				'id' => 13,
 				'key' => 1,
 				'val' => 2
 			)
diff --git a/tests/db_files/FB_TEST_DB.FDB b/tests/db_files/FB_TEST_DB.FDB
index 8de1c33..bf1e569 100755
Binary files a/tests/db_files/FB_TEST_DB.FDB and b/tests/db_files/FB_TEST_DB.FDB differ