diff --git a/tools/ui-components/src/container/container.stories.tsx b/tools/ui-components/src/container/container.stories.tsx
index 5452a04c98a..21b1e0cbce7 100644
--- a/tools/ui-components/src/container/container.stories.tsx
+++ b/tools/ui-components/src/container/container.stories.tsx
@@ -4,7 +4,14 @@ import { Container, ContainerProps } from '.';
 
 const story = {
   title: 'Example/Container',
-  component: Container
+  component: Container,
+  argTypes: {
+    fluid: {
+      control: {
+        type: 'boolean'
+      }
+    }
+  }
 };
 
 const Template: Story<ContainerProps> = args => {
@@ -29,9 +36,4 @@ const Template: Story<ContainerProps> = args => {
 export const Default = Template.bind({});
 Default.args = {};
 
-export const Fluid = Template.bind({});
-Fluid.args = {
-  fluid: true
-};
-
 export default story;