diff --git a/templates/media/media-with-message.php b/templates/media/media-with-message.php
index 09900cb45..865156eca 100644
--- a/templates/media/media-with-message.php
+++ b/templates/media/media-with-message.php
@@ -77,6 +77,8 @@ function( response )
 				'is_up_shortcode' => false,
 				'allow_anonymous' => true,
 				'privacy_enabled' => false,
+				'context'         => 'message-media',
+				'context_id'      => get_current_user_id(),
 			]
 		);
 			?>