diff --git a/packages/badge/test/demo.spec.js b/packages/badge/test/demo.spec.js
new file mode 100644
index 000000000..d647cfabc
--- /dev/null
+++ b/packages/badge/test/demo.spec.js
@@ -0,0 +1,4 @@
+import Demo from '../demo';
+import demoTest from '../../../test/demo-test';
+
+demoTest(Demo);
diff --git a/packages/badge/test/index.spec.js b/packages/badge/test/index.spec.js
index de93a1a94..9b3afe957 100644
--- a/packages/badge/test/index.spec.js
+++ b/packages/badge/test/index.spec.js
@@ -1,9 +1,6 @@
import { mount } from '@vue/test-utils';
import Badge from '../';
import BadgeGroup from '../../badge-group';
-import demoTest from '../../../test/demo-test';
-
-demoTest(Badge);
test('click', () => {
const onClick = jest.fn();
diff --git a/packages/button/test/__snapshots__/index.spec.js.snap b/packages/button/test/__snapshots__/demo.spec.js.snap
similarity index 98%
rename from packages/button/test/__snapshots__/index.spec.js.snap
rename to packages/button/test/__snapshots__/demo.spec.js.snap
index 850f0f1bf..7cf19b217 100644
--- a/packages/button/test/__snapshots__/index.spec.js.snap
+++ b/packages/button/test/__snapshots__/demo.spec.js.snap
@@ -1,6 +1,6 @@
// Jest Snapshot v1, https://goo.gl/fbAQLP
-exports[`renders button correctly 1`] = `
+exports[`renders demo correctly 1`] = `