Null Ref fix