Fixed error that occurs when you don't generate masks