diff --git a/includes/Classifai/Features/ImageGeneration.php b/includes/Classifai/Features/ImageGeneration.php
index a55f39870..b617dc596 100644
--- a/includes/Classifai/Features/ImageGeneration.php
+++ b/includes/Classifai/Features/ImageGeneration.php
@@ -137,7 +137,7 @@ public function register_generate_media_page() {
 			$number_of_images > 1 ? esc_html__( 'Generate Images', 'classifai' ) : esc_html__( 'Generate Image', 'classifai' ),
 			$number_of_images > 1 ? esc_html__( 'Generate Images', 'classifai' ) : esc_html__( 'Generate Image', 'classifai' ),
 			'upload_files',
-			esc_url( admin_url( 'upload.php?action=classifai-generate-image' ) ),
+			esc_url( admin_url( 'upload.php?action=classifai-generate-image&mode=grid' ) ),
 			''
 		);
 	}