Fixed possible ArgumentNullException