diff --git a/TelegramUI/ThemeGridSearchContentNode.swift b/TelegramUI/ThemeGridSearchContentNode.swift
index f7ce7af7e9..03d162e969 100644
--- a/TelegramUI/ThemeGridSearchContentNode.swift
+++ b/TelegramUI/ThemeGridSearchContentNode.swift
@@ -532,7 +532,7 @@ final class ThemeGridSearchContentNode: SearchDisplayControllerContentNode {
                                     i += 1
                                 }
                                 updateSearchContext { _ in
-                                    return (ThemeGridSearchContext(result: ThemeGridSearchResult(query: query, collection: collection, items: collection.results, nextOffset: collection.nextOffset), loadMoreIndex: nil), true)
+                                    return (ThemeGridSearchContext(result: ThemeGridSearchResult(query: wallpaperQuery, collection: collection, items: collection.results, nextOffset: collection.nextOffset), loadMoreIndex: nil), true)
                                 }
                                 return (entries, false)
                             }