diff --git a/web/src/components/common/O2AIContextAddBtn.spec.js b/web/src/components/common/O2AIContextAddBtn.spec.js
index 16511cdbd3..cccd0e4bd9 100644
--- a/web/src/components/common/O2AIContextAddBtn.spec.js
+++ b/web/src/components/common/O2AIContextAddBtn.spec.js
@@ -116,6 +116,6 @@ describe('O2AIContextAddBtn', () => {
it('has correct button attributes', () => {
const button = wrapper.find('[data-test="o2-ai-context-add-btn"]');
expect(button.attributes('flat')).toBe(undefined);
- expect(button.attributes('style')).toContain('border-radius: 100%');
+ expect(button.classes()).toContain('o2-ai-context-btn');
});
});
diff --git a/web/src/components/common/O2AIContextAddBtn.vue b/web/src/components/common/O2AIContextAddBtn.vue
index d0f00ce97a..0fab57b8d3 100644
--- a/web/src/components/common/O2AIContextAddBtn.vue
+++ b/web/src/components/common/O2AIContextAddBtn.vue
@@ -3,17 +3,14 @@
+