From 930f4b350474435e011b9dca18424dd1c42ea353 Mon Sep 17 00:00:00 2001
From: Joshua Hill
Date: Mon, 21 Jun 2010 03:47:54 -0400
Subject: Finally fixed the out of control problem

---
 src/restore.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/restore.c')

diff --git a/src/restore.c b/src/restore.c
index b4cf0b2..bf4b62f 100644
--- a/src/restore.c
+++ b/src/restore.c
@@ -193,7 +193,7 @@ int restore_reboot(struct idevicerestore_client_t* client) {
 		}
 	}
 
-	restore_error = restored_reboot(client);
+	restore_error = restored_reboot(client->restore->client);
 	if (restore_error != RESTORE_E_SUCCESS) {
 		error("ERROR: Unable to reboot the device from restore mode\n");
 		return -1;
@@ -590,7 +590,7 @@ int restore_device(struct idevicerestore_client_t* client, const char* uuid, con
 	restore_error = restored_start_restore(restore);
 	if (restore_error != RESTORE_E_SUCCESS) {
 		error("ERROR: Unable to start the restore process\n");
-		restore_close(device, restore);
+		restore_client_free(client);
 		return -1;
 	}
 
@@ -652,6 +652,6 @@ int restore_device(struct idevicerestore_client_t* client, const char* uuid, con
 		message = NULL;
 	}
 
-	restore_close(device, restore);
+	restore_client_free(client);
 	return 0;
 }
-- 
cgit v1.1-32-gdbae