Fix progress bar for DFU/Recovery image upload

1 files changed, 2 insertions, 1 deletions

diff --git a/src/dfu.c b/src/dfu.c
index 4650e9a..6ba23e7 100644
--- a/src/dfu.c
+++ b/src/dfu.c
@@ -61,7 +61,6 @@ int dfu_client_new(struct idevicerestore_client_t* client)
 
 	irecv_event_subscribe(dfu, IRECV_PROGRESS, &dfu_progress_callback, NULL);
 	client->dfu->client = dfu;
-	register_progress('DFUP', "Uploading");
 	return 0;
 }
 
@@ -200,7 +199,9 @@ int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_ide
 
 	logger(LL_INFO, "Sending %s (%zu bytes)...\n", component, size);
 
+	register_progress('DFUP', "Uploading");
 	irecv_error_t err = irecv_send_buffer(client->dfu->client, data, size, IRECV_SEND_OPT_DFU_NOTIFY_FINISH);
+	finalize_progress('DFUP');
 	if (err != IRECV_E_SUCCESS) {
 		logger(LL_ERROR, "Unable to send %s component: %s\n", component, irecv_strerror(err));
 		free(data);