fix wait_ms function