diff --git a/DRAMSys/traceAnalyzer/presentation/traceplot.cpp b/DRAMSys/traceAnalyzer/presentation/traceplot.cpp
index 86c5869f..07243e1c 100644
--- a/DRAMSys/traceAnalyzer/presentation/traceplot.cpp
+++ b/DRAMSys/traceAnalyzer/presentation/traceplot.cpp
@@ -151,6 +151,7 @@ void TracePlot::setUpActions()
                      SLOT(on_colorGroupingTransaction()));
 
     setColorGroupingRainbowTransaction = new QAction("Group by Transaction - Rainbow Colored", this);
+    setColorGroupingRainbowTransaction->setCheckable(true);
     addAction(setColorGroupingRainbowTransaction);
     QObject::connect(setColorGroupingRainbowTransaction, SIGNAL(triggered()), this,
                      SLOT(on_colorGroupingRainbowTransaction()));
@@ -164,6 +165,7 @@ void TracePlot::setUpActions()
     QActionGroup *colorGroupingGroup = new QActionGroup(this);
     colorGroupingGroup->addAction(setColorGroupingPhase);
     colorGroupingGroup->addAction(setColorGroupingTransaction);
+    colorGroupingGroup->addAction(setColorGroupingRainbowTransaction);
     colorGroupingGroup->addAction(setColorGroupingThread);
 
     exportToPdf = new QAction("Export to SVG", this);