Fix a None error