diff --git a/tests/qunit/index.html b/tests/qunit/index.html
index 81244c51a7..9b05fb6f78 100644
--- a/tests/qunit/index.html
+++ b/tests/qunit/index.html
@@ -24,6 +24,8 @@
 			};
 		</script>
 		<script src="../../build/wp-includes/js/wp-util.js"></script>
+		<script src="../../build/wp-includes/js/dist/vendor/wp-polyfill.js"></script>
+		<script src="../../build/wp-includes/js/dist/dom-ready.js"></script>
 		<script src="../../build/wp-includes/js/dist/a11y.js"></script>
 		<script>
 			window._wpMediaModelsL10n = {"settings":{"ajaxurl":"\/wp-admin\/admin-ajax.php","post":{"id":0}}};
diff --git a/tests/qunit/wp-admin/js/customize-controls.js b/tests/qunit/wp-admin/js/customize-controls.js
index 47ab5c0203..ed92f5a2cd 100644
--- a/tests/qunit/wp-admin/js/customize-controls.js
+++ b/tests/qunit/wp-admin/js/customize-controls.js
@@ -94,13 +94,14 @@ jQuery( window ).load( function (){
 		collection = new wp.customize.Notifications({
 			container: container
 		});
-		collection.add( 'mycode-1', new wp.customize.Notification( 'mycode-1' ) );
+		collection.add( 'mycode-1', new wp.customize.Notification( 'mycode-1', { message: 'My message 1' } ) );
 		collection.render();
 		items = collection.container.find( 'li' );
 		equal( items.length, 1 );
 		equal( items.first().data( 'code' ), 'mycode-1' );
 
 		collection.add( 'mycode-2', new wp.customize.Notification( 'mycode-2', {
+			message: 'My message 2',
 			dismissible: true
 		} ) );
 		collection.render();
@@ -187,8 +188,8 @@ jQuery( window ).load( function (){
 			assert.equal( 0, notificationContainerElement.find( '> ul > li' ).length );
 			assert.equal( 0, notificationContainerElement.height() );
 
-			settingNotification = new wp.customize.Notification( 'setting_invalidity', 'Invalid setting' );
-			controlOnlyNotification = new wp.customize.Notification( 'control_invalidity', 'Invalid control' );
+			settingNotification = new wp.customize.Notification( 'setting_invalidity', { message: 'Invalid setting' } );
+			controlOnlyNotification = new wp.customize.Notification( 'control_invalidity', { message: 'Invalid control' } );
 			control.settings['default'].notifications.add( settingNotification.code, settingNotification );
 			control.notifications.add( controlOnlyNotification.code, controlOnlyNotification );