fix: restored uncommented "wait" method

Looked to me like the wait method had been mistakenly commented out?