diff --git a/ui/src/features/project/pipelines/images.tsx b/ui/src/features/project/pipelines/images.tsx
index 4e8bb3cfc..93ca35651 100644
--- a/ui/src/features/project/pipelines/images.tsx
+++ b/ui/src/features/project/pipelines/images.tsx
@@ -113,7 +113,7 @@ export const Images = ({
   stages: Stage[];
   hide: () => void;
 }) => {
-  const colors = useContext(ColorContext);
+  const { stageColorMap: colors } = useContext(ColorContext);
   const images = useMemo(() => {
     const images = new Map<string, Map<string, StageStyleMap>>();
     stages.forEach((stage) => {