diff --git a/core/specification/src/stories/controls-button.stories.js b/core/specification/src/stories/controls-button.stories.js
index 80eaaa075..d35612c9d 100644
--- a/core/specification/src/stories/controls-button.stories.js
+++ b/core/specification/src/stories/controls-button.stories.js
@@ -1,5 +1,5 @@
 import React from 'react';
-import { Button } from '@theme-ui/components';
+import { Button } from 'theme-ui';
 import { ControlTypes } from '@component-controls/specification';
 import { ButtonControl } from './components/ButtonControl';
 
diff --git a/core/specification/src/stories/controls-object.stories.js b/core/specification/src/stories/controls-object.stories.js
index 89e826976..e6b2188fe 100644
--- a/core/specification/src/stories/controls-object.stories.js
+++ b/core/specification/src/stories/controls-object.stories.js
@@ -1,6 +1,6 @@
 import React from 'react';
 import { ControlTypes } from '@component-controls/specification';
-import { Button } from '@theme-ui/components';
+import { Button } from 'theme-ui';
 import { ObjectControl } from './components/ObjectControl';
 
 export default {