Fix an minor crash issue