Fix comparison to None