Fix bug in `run_forever`