Update after changing the mimicry of feat.common.time.wait_for() function.