1 files changed, 2 insertions, 1 deletions

diff --git a/apps/playlist_viewer.c b/apps/playlist_viewer.c
index e3faf4b68f..5b6e2991bc 100644
--- a/apps/playlist_viewer.c
+++ b/apps/playlist_viewer.c
@@ -883,7 +883,8 @@ enum playlist_viewer_result playlist_viewer_ex(const char* filename)
                 else
                 {
                     exit = true;
-                    ret = PLAYLIST_VIEWER_CANCEL;
+                    ret = button == ACTION_TREE_WPS ?
+                                    PLAYLIST_VIEWER_OK : PLAYLIST_VIEWER_CANCEL;
                 }
                 break;
             }