Add back tzinfo