diff --git a/src/wp-admin/js/updates.js b/src/wp-admin/js/updates.js
index cf63c17e2b..b62c1de869 100644
--- a/src/wp-admin/js/updates.js
+++ b/src/wp-admin/js/updates.js
@@ -1193,7 +1193,7 @@
 				$button = $( '.theme-install[data-slug="' + response.slug + '"]' );
 				$card   = $button.closest( '.theme' ).addClass( 'theme-install-failed' ).append( $message );
 			}
-			$( '.wp-full-overlay' ).removeClass( 'customize-loading' );
+			wp.customize.notifications.remove( 'theme_installing' );
 		} else {
 			if ( $document.find( 'body' ).hasClass( 'full-overlay-active' ) ) {
 				$button = $( '.theme-install[data-slug="' + response.slug + '"]' );