fixed null ref error