Fix imports in time unit tests.