Change don't use == None