Fix xyt axis = None error