Null ref fix