1 files changed, 2 insertions, 4 deletions

diff --git a/dev/afccheck.c b/dev/afccheck.c
index 88935a0..569acf1 100644
--- a/dev/afccheck.c
+++ b/dev/afccheck.c
@@ -93,15 +93,13 @@ int main(int argc, char *argv[])
         lockdownd_client_t client = NULL;
         iphone_device_t phone = NULL;
         GError *err;
-        int port = 0;
+        uint16_t port = 0;
         afc_client_t afc = NULL;
 
         if (argc > 1 && !strcasecmp(argv[1], "--debug")) {
                 iphone_set_debug_level(1);
-                iphone_set_debug_mask(DBGMASK_ALL);
         } else {
                 iphone_set_debug_level(0);
-                iphone_set_debug_mask(DBGMASK_NONE);
         }
 
         if (IPHONE_E_SUCCESS != iphone_device_new(&phone, NULL)) {
@@ -109,7 +107,7 @@ int main(int argc, char *argv[])
                 return 1;
         }
 
-        if (LOCKDOWN_E_SUCCESS != lockdownd_client_new(phone, &client)) {
+        if (LOCKDOWN_E_SUCCESS != lockdownd_client_new_with_handshake(phone, &client, "afccheck")) {
                 iphone_device_free(phone);
                 return 1;
         }