Fix 'wait' block to have input and output.