diff --git a/docs/reference/createPlugin-feature-flags.md b/docs/reference/createPlugin-feature-flags.md
index 60fddaa8fd..ebd81bcb20 100644
--- a/docs/reference/createPlugin-feature-flags.md
+++ b/docs/reference/createPlugin-feature-flags.md
@@ -45,7 +45,7 @@ Then, later, you can directly use it via `useApi`:
 ```tsx
 import React, { FC } from 'react';
 import { Button } from '@material-ui/core';
-import { featureFlagsApiRef, useApi } from '@backstage/core';
+import { FeatureFlagState, featureFlagsApiRef, useApi } from '@backstage/core';
 
 const ExampleButton: FC<{}> = () => {
   const flags = useApi(featureFlagsApiRef).getFlags();