gemini.suite("IconButton", () => {
  gemini.suite("Default", suite => {
    suite
      .setUrl("iframe.html?selectedKind=IconButton&selectedStory=default")
      .setCaptureElements(".hig__icon-button")
      .capture("Default");
  });

  gemini.suite("with link", suite => {
    suite
      .setUrl("iframe.html?selectedKind=IconButton&selectedStory=with%20link")
      .setCaptureElements(".hig__icon-button")
      .capture("with link");
  });
});

@@ -178,32 +178,35 @@
 ments(%22.
-hig__icon-button
+storybook-component
 %22)%0A     
@@ -394,24 +394,27 @@
 s(%22.
-hig__icon-button
+storybook-component
 %22)%0A 

update gemini capture element