1 files changed, 5 insertions, 4 deletions
diff --git a/examples/barchart/main.cpp b/examples/barchart/main.cpp
index 037fa214..9f17f8d9 100644
--- a/examples/barchart/main.cpp
+++ b/examples/barchart/main.cpp
@@ -236,10 +236,11 @@ int main(int argc, char **argv)
     layout->addWidget(container, 1);
     layout->addWidget(tableWidget, 1, Qt::AlignHCenter);
 
-    // We don't need to initialize the mapping object in any way, as it defaults
-    // to row/column support and uses the Qt::DisplayRole role for value role by default.
-    QItemModelBarDataProxy *proxy = new QItemModelBarDataProxy(tableWidget->model(),
-                                                               new QItemModelBarDataMapping);
+    // Since we are dealing with QTableWidget, the model will already have data sorted properly
+    // in rows and columns, so create a custom mapping to utilize this.
+    QItemModelBarDataMapping *mapping = new QItemModelBarDataMapping;
+    mapping->setUseModelCategories(true);
+    QItemModelBarDataProxy *proxy = new QItemModelBarDataProxy(tableWidget->model(), mapping);
     chart->setActiveDataProxy(proxy);
 
     ChartDataGenerator *generator = new ChartDataGenerator(chart, tableWidget);