Using wait_for_message instead of subscribing then publishing