[Minor] Null check fixed