diff --git a/colorLabels.m b/colorLabels.m
index f8b7029b1aa55f6de5301d429705a287cd7040c8..b61a0479cad90bf9e3ca7c290605cf4213ec7f10 100644
--- a/colorLabels.m
+++ b/colorLabels.m
@@ -2,9 +2,10 @@ function labels = colorLabels(labels,varargin)
 %COLORLABELS Summary of this function goes here
 %   Detailed explanation goes here
 
-cmap = jet(max(labels(:)));
+sz = size(labels);
+cmap = jet(max(labels(:))+1);
 shuffle = false;
-gray = ones(size(labels));
+gray = ones(sz);
 coldim = 3;
 
 p = inputParser();
@@ -31,8 +32,8 @@ order = 1:(ndims(gray)+1);
 order(coldim) = ndims(gray)+1;
 order(ndims(gray)+1) = coldim;
 
-labels = cmap(labels(:),:);
-labels = reshape(labels,[size(gray),3]);
+labels = cmap(labels(:)+1,:);
+labels = reshape(labels,[sz,3]);
 labels = permute(labels,order);
 labels = labels.*gray;