Add missing `import time`