Updated imports to make it easier to mock time from another package.