diff --git a/align.m b/align.m
index c20e467a21362012e631f605c9a77f9d34d65c23..27c6b7ce253cdbb24a4e37a3a724f3db155681a4 100644
--- a/align.m
+++ b/align.m
@@ -21,12 +21,14 @@ for lab = 1:numLab
     
     if ~isempty(Col1{lab})
         LFSize1 = size(Col1{lab});
+        LFSize1(end+1:5) = 1;
         ImgSize1{lab} = LFSize1(1:3);
         ImgRes1 {lab} = LFSize1(4:end);
     end
     
     if ~isempty(Col2{lab})
         LFSize2 = size(Col2{lab});
+        LFSize2(end+1:5) = 1;
         ImgSize2{lab} = LFSize2(1:3);
         ImgRes2 {lab} = LFSize2(4:end);
     end
diff --git a/join.m b/join.m
index b8eb5c14970e3dd6066c0b58697f81b48fbffe1b..9c3d405a23118695c4489d8133bb87a39fdbc324 100644
--- a/join.m
+++ b/join.m
@@ -7,6 +7,7 @@ function Set = join(Set)
 [Color,Label,Offset] = utils.join(Color,Label,Offset);
 
 LFSize = size(Color);
+LFSize(end+1:5) = 1;
 ImgSize = LFSize(1:3);
 ImgRes = LFSize(4:end);