diff --git a/src/gs-updates-section.c b/src/gs-updates-section.c
index 8867da797752cadf82d57326547f01ffe1cf7614..c3a835198ffa9032815603764b1924d32054fc1a 100644
--- a/src/gs-updates-section.c
+++ b/src/gs-updates-section.c
@@ -323,13 +323,13 @@ _update_buttons (GsUpdatesSection *self)
 		gtk_widget_show (GTK_WIDGET (self->button_stack));
 		/* TRANSLATORS: This is the button for installing all
 		 * offline updates */
-		gtk_button_set_label (GTK_BUTTON (self->button_update), _("Restart & Update"));
+		gs_progress_button_set_label (GS_PROGRESS_BUTTON (self->button_update), _("Restart & Update"));
 	} else if (self->kind == GS_UPDATES_SECTION_KIND_ONLINE) {
 		gtk_stack_set_visible_child_name (GTK_STACK (self->button_stack), "update");
 		gtk_widget_show (GTK_WIDGET (self->button_stack));
 		/* TRANSLATORS: This is the button for upgrading all
 		 * online-updatable applications */
-		gtk_button_set_label (GTK_BUTTON (self->button_update), _("Update All"));
+		gs_progress_button_set_label (GS_PROGRESS_BUTTON (self->button_update), _("Update All"));
 	} else {
 		gtk_widget_hide (GTK_WIDGET (self->button_stack));
 	}