diff --git a/core/ui/src/main/kotlin/dev/meloda/fast/ui/components/FullScreenDialog.kt b/core/ui/src/main/kotlin/dev/meloda/fast/ui/components/FullScreenDialog.kt
index 12ae5563..e6cea9ac 100644
--- a/core/ui/src/main/kotlin/dev/meloda/fast/ui/components/FullScreenDialog.kt
+++ b/core/ui/src/main/kotlin/dev/meloda/fast/ui/components/FullScreenDialog.kt
@@ -12,13 +12,14 @@ import androidx.compose.ui.zIndex
 @Composable
 fun FullScreenDialog(
     modifier: Modifier = Modifier,
-    content: @Composable () -> Unit
+    onDismiss: () -> Unit = {},
+    content: @Composable () -> Unit,
 ) {
     Dialog(
-        onDismissRequest = {},
-        DialogProperties(
+        onDismissRequest = onDismiss,
+        properties = DialogProperties(
             usePlatformDefaultWidth = false,
-            decorFitsSystemWindows = false
+            decorFitsSystemWindows = false,
         )
     ) {
         Box(
diff --git a/feature/photoviewer/src/main/kotlin/dev/meloda/fast/photoviewer/presentation/PhotoViewScreen.kt b/feature/photoviewer/src/main/kotlin/dev/meloda/fast/photoviewer/presentation/PhotoViewScreen.kt
index 17681955..7c69c4ac 100644
--- a/feature/photoviewer/src/main/kotlin/dev/meloda/fast/photoviewer/presentation/PhotoViewScreen.kt
+++ b/feature/photoviewer/src/main/kotlin/dev/meloda/fast/photoviewer/presentation/PhotoViewScreen.kt
@@ -80,7 +80,10 @@ fun PhotoViewDialog(
     val applicationContext = LocalContext.current.applicationContext
 
     if (photoViewerInfo != null) {
-        FullScreenDialog(modifier = modifier) {
+        FullScreenDialog(
+            modifier = modifier,
+            onDismiss = onDismiss
+        ) {
             val viewModel = remember(true) {
                 PhotoViewViewModelImpl(
                     arguments = PhotoViewArguments(