Fix None error