1 files changed, 3 insertions, 3 deletions

diff --git a/src/diagnostics_relay.c b/src/diagnostics_relay.c
index 0834700..3ac6058 100644
--- a/src/diagnostics_relay.c
+++ b/src/diagnostics_relay.c
@@ -229,7 +229,7 @@ LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_sleep(diagnosti
         return ret;
 }
 
-static diagnostics_relay_error_t internal_diagnostics_relay_action(diagnostics_relay_client_t client, const char* name, int flags)
+static diagnostics_relay_error_t internal_diagnostics_relay_action(diagnostics_relay_client_t client, const char* name, diagnostics_relay_action_t flags)
 {
         if (!client)
                 return DIAGNOSTICS_RELAY_E_INVALID_ARG;
@@ -273,12 +273,12 @@ static diagnostics_relay_error_t internal_diagnostics_relay_action(diagnostics_r
         return ret;
 }
 
-LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_restart(diagnostics_relay_client_t client, int flags)
+LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_restart(diagnostics_relay_client_t client, diagnostics_relay_action_t flags)
 {
         return internal_diagnostics_relay_action(client, "Restart", flags);
 }
 
-LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_shutdown(diagnostics_relay_client_t client, int flags)
+LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_shutdown(diagnostics_relay_client_t client, diagnostics_relay_action_t flags)
 {
         return internal_diagnostics_relay_action(client, "Shutdown", flags);
 }