Can this be reduced to `exit_code = container.wait()`?