Align correct wait_callback state.