1 files changed, 3 insertions, 2 deletions
diff --git a/plugins/UserSettings/API.php b/plugins/UserSettings/API.php
index 7a9f5b6dda..c1c5535c1c 100644
--- a/plugins/UserSettings/API.php
+++ b/plugins/UserSettings/API.php
@@ -11,6 +11,7 @@
 use Piwik\Archive;
 use Piwik\Metrics;
 use Piwik\Piwik;
+use Piwik\DataTable;
 
 /**
  * @see plugins/UserSettings/functions.php
@@ -111,7 +112,7 @@ class Piwik_UserSettings_API
         );
 
         $dataTables = array($dataTable);
-        if ($dataTable instanceof Piwik_DataTable_Array) {
+        if ($dataTable instanceof DataTable\Map) {
             $dataTables = $dataTable->getArray();
         }
         foreach ($dataTables AS $table) {
@@ -180,7 +181,7 @@ class Piwik_UserSettings_API
         $visitsSums = $archive->getDataTableFromNumeric('nb_visits');
 
         // check whether given tables are arrays
-        if ($dataTable instanceof Piwik_DataTable_Array) {
+        if ($dataTable instanceof DataTable\Map) {
             $tableArray = $dataTable->getArray();
             $browserTypesArray = $browserTypes->getArray();
             $visitSumsArray = $visitsSums->getArray();