diff --git a/write.m b/write.m
index 79f07433e2ba006fe6e1349705cbc1487c0a0899..566af799fa470ee512c28d4cae7fd44ff732c045 100644
--- a/write.m
+++ b/write.m
@@ -13,6 +13,7 @@ p.parse(varargin{:});
 %filename = p.Results.filename;
 
 LFSize = size(LFRef);
+LFSize(end+1:5) = 1;
 imgSize = LFSize(1:3);
 imgRes = LFSize(4:end);