Merge branch 'tonic-management-interface'

1 files changed, 6 insertions, 5 deletions

diff --git a/mullvad-cli/src/cmds/beta_program.rs b/mullvad-cli/src/cmds/beta_program.rs
index c4cbe6cf21..73a0ba15c7 100644
--- a/mullvad-cli/src/cmds/beta_program.rs
+++ b/mullvad-cli/src/cmds/beta_program.rs
@@ -3,6 +3,7 @@ use clap::value_t_or_exit;
 
 pub struct BetaProgram;
 
+#[mullvad_management_interface::async_trait]
 impl Command for BetaProgram {
     fn name(&self) -> &'static str {
         "beta-program"
@@ -24,11 +25,11 @@ impl Command for BetaProgram {
             .subcommand(clap::SubCommand::with_name("get").about("Get beta notifications setting"))
     }
 
-    fn run(&self, matches: &clap::ArgMatches<'_>) -> Result<()> {
+    async fn run(&self, matches: &clap::ArgMatches<'_>) -> Result<()> {
         match matches.subcommand() {
             ("get", Some(_)) => {
-                let mut rpc = new_rpc_client()?;
-                let settings = rpc.get_settings()?;
+                let mut rpc = new_rpc_client().await?;
+                let settings = rpc.get_settings(()).await?.into_inner();
                 let enabled_str = if settings.show_beta_releases {
                     "on"
                 } else {
@@ -47,8 +48,8 @@ impl Command for BetaProgram {
                     ));
                 }
 
-                let mut rpc = new_rpc_client()?;
-                rpc.set_show_beta_releases(enable)?;
+                let mut rpc = new_rpc_client().await?;
+                rpc.set_show_beta_releases(enable).await?;
 
                 println!("Beta program: {}", enable_str);
                 Ok(())