diff --git a/src/layouts/bailout-page/bailout-page.spec.js b/src/layouts/bailout-page/bailout-page.spec.js
index 3ccd149b..83841efb 100644
--- a/src/layouts/bailout-page/bailout-page.spec.js
+++ b/src/layouts/bailout-page/bailout-page.spec.js
@@ -211,6 +211,27 @@ describe('Bailout page', () => {
                 // Assert
                 expect(name).toBe(expected);
             });
+            test('returns YMMNotFound widget name when bailout code is BailoutCode.YMMNotFound', () => {
+                // Arrange
+                const mainInitialState = {
+                    applicationUser: {
+                        pageData: {
+                            'bailout-page': {
+                                bailoutCode: bailoutCode.YMMNotFound
+                            }
+                        }
+                    }
+                };
+                const initialData = { };
+                const { wrapper } = getMountedComponent(mainInitialState, initialData);
+                const expected = 'YMMNotFoundWidget';
+
+                // Act
+                const name = wrapper.vm.subHeaderCmsWidgetName;
+
+                // Assert
+                expect(name).toBe(expected);
+            });
             test('returns default widget name when bailoutCode does not match any other content bailout codes', () => {
                 // Arrange
                 const mainInitialState = {