Add missing import on pytz