Good point. I'll add a check