diff --git a/resources/classes/schema.php b/resources/classes/schema.php
index be7f10dbb5..b98f778206 100644
--- a/resources/classes/schema.php
+++ b/resources/classes/schema.php
@@ -72,10 +72,10 @@ include "root.php";
 										else {
 											$sql .= $field['type'];
 										}
-										if ($field['key'] == "primary") {
+										if ($field['key']['type'] == "primary") {
 											$sql .= " PRIMARY KEY";
 										}
-										if ($field['key'] == "foreign") {
+										if ($field['key']['type'] == "foreign") {
 											if ($this->db_type == "pgsql") {
 												//$sql .= " references ".$field['key']['reference']['table']."(".$field['key']['reference']['field'].")";
 											}
diff --git a/resources/schema.php b/resources/schema.php
index 3687065496..f9ea1681f0 100644
--- a/resources/schema.php
+++ b/resources/schema.php
@@ -188,7 +188,7 @@ function db_create_table ($apps, $db_type, $table) {
 						else {
 							$sql .= $field['type'];
 						}
-						if ($field['key'] == "primary") {
+						if ($field['key']['type'] == "primary") {
 							$sql .= " PRIMARY KEY";
 						}
 						$field_count++;