diff --git a/src/containers/HSM/HSM.test.tsx b/src/containers/HSM/HSM.test.tsx
index d926a138b..a1a2b89a9 100644
--- a/src/containers/HSM/HSM.test.tsx
+++ b/src/containers/HSM/HSM.test.tsx
@@ -147,6 +147,12 @@ describe('Add mode', () => {
     });
 
     fireEvent.click(screen.getByText('Add Variable'));
+    fireEvent.click(screen.getByText('Add buttons'));
+
+    fireEvent.change(screen.getByPlaceholderText('Button Title'), { target: { value: 'Call me' } });
+    fireEvent.change(screen.getByPlaceholderText('Button Value'), {
+      target: { value: '9876543210' },
+    });
 
     await waitFor(() => {
       expect(screen.getByText('Hi, How are you {{1}}')).toBeInTheDocument();
diff --git a/src/mocks/Template.tsx b/src/mocks/Template.tsx
index 9e6bb62c5..888ca7fa1 100644
--- a/src/mocks/Template.tsx
+++ b/src/mocks/Template.tsx
@@ -280,6 +280,9 @@ export const templateMock = createTemplateMock({
   languageId: '1',
   example: 'Hi, How are you [User]',
   shortcode: 'element_name',
+  hasButtons: true,
+  buttons: '[{"type":"PHONE_NUMBER","text":"Call me","phone_number":"9876543210"}]',
+  buttonType: 'CALL_TO_ACTION',
 });
 
 export const quickReplyMock = createTemplateMock({
@@ -295,7 +298,7 @@ export const quickReplyMock = createTemplateMock({
   example: 'Hi',
   shortcode: 'welcome',
   hasButtons: true,
-  buttons: '[{"type":"QUICK_REPLY","text":""},{"type":"QUICK_REPLY","text":""}]',
+  buttons: '[{"type":"QUICK_REPLY","text":"Yes"},{"type":"QUICK_REPLY","text":"No"}]',
   buttonType: 'QUICK_REPLY',
 });
 
@@ -312,7 +315,7 @@ export const ctaMock = createTemplateMock({
   example: 'Hi',
   shortcode: 'welcome',
   hasButtons: true,
-  buttons: '[{"type":"PHONE_NUMBER","text":"","phone_number":""}]',
+  buttons: '[{"type":"PHONE_NUMBER","text":"Call me","phone_number":"9876543210"}]',
   buttonType: 'CALL_TO_ACTION',
 });