Fix VisitByServerTime api method to handle Datatable Maps (#11619)

1 files changed, 7 insertions, 1 deletions

diff --git a/plugins/VisitTime/API.php b/plugins/VisitTime/API.php
index f706860b4e..28b73b99a4 100644
--- a/plugins/VisitTime/API.php
+++ b/plugins/VisitTime/API.php
@@ -54,7 +54,13 @@ class API extends \Piwik\Plugin\API
         $table->filter('Piwik\Plugins\VisitTime\DataTable\Filter\AddSegmentByLabelInUTC', array($timezone, $period, $date));
 
         if ($hideFutureHoursWhenToday) {
-            $table = $this->removeHoursInFuture($table, $idSite, $period, $date);
+            if ($table instanceof DataTable\Map) {
+                foreach ($table->getDataTables() as &$dataTable) {
+                    $dataTable = $this->removeHoursInFuture($dataTable, $idSite, $period, $date);
+                }
+            } else {
+                $table = $this->removeHoursInFuture($table, $idSite, $period, $date);
+            }
         }
 
         return $table;