Fix NULL check