Update wait_msg.c