Fix ArgumentNullException