diff --git a/factorize.m b/factorize.m
index 447434be4457f55145c42c631881dc9ab9e97075..f85d031a8815acfb2a87260207337bfe31b8d344 100644
--- a/factorize.m
+++ b/factorize.m
@@ -1,8 +1,8 @@
-function [B,C,U,S,V] = factorize(M,k)
+function [B,C,U,S,V] = factorize(M,k,Mask)
 %FACTORIZE Summary of this function goes here
 %   Detailed explanation goes here
 
-Mask = isnan(M);
+if ~exist('Mask','var'); Mask = isnan(M); end
 M(Mask) = 0;
 
 [U,S,V] = utils.svd_nsq(M);