Да. Вероятно, теперь стоит убрать wait_for, раз каких-то еще действий не предполагается делать.