Update if statement in time.rs