Fix NULL edge cases