diff --git a/html/install.php b/html/install.php
index 6aac6bc5d4..123fac346d 100644
--- a/html/install.php
+++ b/html/install.php
@@ -307,6 +307,7 @@ elseif($stage == "2") {
     $config['db_user']=$dbuser;
     $config['db_pass']=$dbpass;
     $config['db_name']=$dbname;
+    $config['db']['extension']='mysqli';
     $sql_file = '../build.sql';
     require '../build-base.php';
 ?>
diff --git a/includes/sql-schema/update.php b/includes/sql-schema/update.php
index 6497d1ca55..46a49a8261 100644
--- a/includes/sql-schema/update.php
+++ b/includes/sql-schema/update.php
@@ -109,7 +109,7 @@ foreach ($filelist as $file) {
 
                     if ($line[0] != '#') {
                         if ($config['db']['extension'] == 'mysqli') {
-                            $update = mysqli_query($database_link, $line);
+                            $update = mysqli_query($connection, $line);
                         }
                         else {
                             $update = mysql_query($line);
@@ -118,7 +118,7 @@ foreach ($filelist as $file) {
                             $err++;
                             if ($debug) {
                                 if ($config['db']['extension'] == 'mysqli') {
-                                    echo mysqli_error($database_link)."\n";
+                                    echo mysqli_error($connection)."\n";
                                 }
                                 else {                           
                                     echo mysql_error()."\n";