2 files changed, 19 insertions, 17 deletions

diff --git a/packages/theme/test/index.test.js b/packages/theme/test/index.test.js
deleted file mode 100644
index 3906433c1..000000000
--- a/packages/theme/test/index.test.js
+++ /dev/null
@@ -1,17 +0,0 @@
-const expect = require('expect.js');
-
-const { colorBackground: colorBackgroundDefault } = require('../lib/themes/default');
-const { colorBackground: colorBackgroundDark } = require('../lib/themes/dark');
-const { default: theme } = require('../lib');
-
-describe('Load theme', () => {
-  it('Should load default theme', () => {
-    const { colorBackground } = theme('default');
-    expect(colorBackground).to.be(colorBackgroundDefault);
-  });
-
-  it('Should load dark theme', () => {
-    const { colorBackground } = theme('dark');
-    expect(colorBackground).to.be(colorBackgroundDark);
-  });
-});
diff --git a/packages/theme/test/index.test.ts b/packages/theme/test/index.test.ts
new file mode 100644
index 000000000..7eda4e359
--- /dev/null
+++ b/packages/theme/test/index.test.ts
@@ -0,0 +1,19 @@
+import expect from 'expect.js';
+
+import makeDefaultThemeConfig from '../src/themes/default';
+import makeDarkThemeConfig from '../src/themes/dark';
+import { theme, ThemeType } from '../src';
+
+describe('Load theme', () => {
+  it('Should load default theme', () => {
+    const { colorBackground } = theme('default' as ThemeType);
+    expect(colorBackground).to.be(
+      makeDefaultThemeConfig('default').colorBackground,
+    );
+  });
+
+  it('Should load dark theme', () => {
+    const { colorBackground } = theme('dark' as ThemeType);
+    expect(colorBackground).to.be(makeDarkThemeConfig('dark').colorBackground);
+  });
+});