diff --git a/src/components/EditGrid/EditGrid.js b/src/components/EditGrid/EditGrid.js
index 70c29fc91..48d5a9b11 100644
--- a/src/components/EditGrid/EditGrid.js
+++ b/src/components/EditGrid/EditGrid.js
@@ -3,7 +3,7 @@ import PropTypes from 'prop-types';
import React from 'react';
import {FormattedMessage} from 'react-intl';
-import OFButton from 'components/Button';
+import {OFButton} from 'components/Button';
import FAIcon from 'components/FAIcon';
const DefaultAddButtonLabel = () => (
diff --git a/src/components/EditGrid/EditGrid.stories.js b/src/components/EditGrid/EditGrid.stories.js
index a01bc2ace..e4c5bc035 100644
--- a/src/components/EditGrid/EditGrid.stories.js
+++ b/src/components/EditGrid/EditGrid.stories.js
@@ -1,6 +1,5 @@
-import {Button} from '@utrecht/component-library-react';
-
import Body from 'components/Body';
+import {OFButton} from 'components/Button';
import {EditGrid, EditGridButtonGroup, EditGridItem} from '.';
@@ -18,7 +17,7 @@ export default {
heading="Item 1"
buttons={