Fix comparison with None