Handle `null` values correctly