diff --git a/src/lib/components/TabsContainer.svelte b/src/lib/components/TabsContainer.svelte
index 22925fe..31bbf56 100644
--- a/src/lib/components/TabsContainer.svelte
+++ b/src/lib/components/TabsContainer.svelte
@@ -116,7 +116,7 @@
             <Tabs elem_classes={["gradio-tabs"]}>
                 {#each children.filter(item => item.id !== SHADOW_PLACEHOLDER_ITEM_ID) as item, i(item.id)}
                     {@const tabName = getTabName(container, i)}
-                    <TabItem name={tabName} on:select={() => console.log("tab " + i)}>
+                    <TabItem name={tabName}>
                         <WidgetContainer dragItem={item} zIndex={zIndex+1} {isMobile} />
                     </TabItem>
                 {/each}
diff --git a/src/lib/widgets/GalleryWidget.svelte b/src/lib/widgets/GalleryWidget.svelte
index e611291..ae167b5 100644
--- a/src/lib/widgets/GalleryWidget.svelte
+++ b/src/lib/widgets/GalleryWidget.svelte
@@ -41,7 +41,7 @@
  };
 
  let style: Styles = {
-     grid_cols: [3],
+     grid_cols: [isMobile ? 2 : 3],
      object_fit: "cover",
  }
  let element: HTMLDivElement;
@@ -90,7 +90,7 @@
          thumbs: images.map(i => i.url),
          type: 'popup',
      });
-     mobileLightbox.open()
+     mobileLightbox.open(selected_image)
 
      event.stopPropagation()
  }