fix None error