Fix missing import in time loop