Updating unit test to use timezones