fix waited_for input type