diff --git a/Telegram/BUCK b/Telegram/BUCK
index c4684b313b..0ba80dc35b 100644
--- a/Telegram/BUCK
+++ b/Telegram/BUCK
@@ -315,10 +315,7 @@ apple_bundle(
 apple_binary(
     name = "NotificationServiceBinary",
     srcs = glob([
-        "NotificationService/*.swift",
-    ]),
-    headers = glob([
-       "NotificationService/*.h", 
+        "NotificationService/Sources/*.swift",
     ]),
     configs = notification_service_extension_configs(),
     swift_compiler_flags = [