diff --git a/matlab/plot/plot_radial_transport_and_spacetime.m b/matlab/plot/plot_radial_transport_and_spacetime.m
index 4e2617cd0097199ca6eb2cfc0d5087e40f40b446..8fdcbc3d2cb4250cb560cd2de67a429ea9032e70 100644
--- a/matlab/plot/plot_radial_transport_and_spacetime.m
+++ b/matlab/plot/plot_radial_transport_and_spacetime.m
@@ -27,12 +27,18 @@ mvm = @(x) movmean(x,OPTIONS.NMVA);
     FIGURE.fig = figure; FIGURE.FIGNAME = ['ZF_transport_drphi','_',DATA.params_string]; %set(gcf, 'Position',  [500, 1000, 1000, 600])
     FIGURE.ax1 = subplot(2,1,1,'parent',FIGURE.fig);
     for ia = 1:DATA.inputs.Na
+        if ia == 1
+            a = 'i'; color = 'r';
+        elseif ia == 2
+            a = 'e'; color = 'b';
+        end
+
         plot(mvm(DATA.Ts0D),mvm(DATA.PGAMMA_RI(ia,:)*SCALE),'--',...
-            'color',clr_(max(1,(DATA.grids.Np-1)/2+(ia-1)),:),...
-            'DisplayName',['$\Gamma_x$ ',DATA.paramshort]); hold on;
+            'color',color,...
+            'DisplayName',['$\Gamma_{x,',a,'}$ ',DATA.paramshort]); hold on;
         plot(mvm(DATA.Ts0D),mvm(DATA.HFLUX_X(ia,:)*SCALE),'-',...
-            'color',clr_(max(1,(DATA.grids.Np-1)/2+(ia-1)),:),...
-            'DisplayName',['$Q_x$ ',DATA.paramshort]); hold on;
+            'color',color,...
+            'DisplayName',['$Q_{x,',a,'}$ ',DATA.paramshort]); hold on;
         ylabel('Transport')  
         if(~isnan(Qx_infty_avg))
         plot(DATA.Ts0D(its0D:ite0D),ones(ite0D-its0D+1,1)*Qx_avg, '-k',...