diff --git a/MTNetworkUsageManager.h b/MTNetworkUsageManager.h
index 8fef54e0f9..a768085655 100644
--- a/MTNetworkUsageManager.h
+++ b/MTNetworkUsageManager.h
@@ -27,6 +27,9 @@ typedef struct {
 - (void)addIncomingBytes:(NSUInteger)incomingBytes interface:(MTNetworkUsageManagerInterface)interface;
 - (void)addOutgoingBytes:(NSUInteger)outgoingBytes interface:(MTNetworkUsageManagerInterface)interface;
 
+- (void)resetIncomingBytes:(MTNetworkUsageManagerInterface)interface;
+- (void)resetOutgoingBytes:(MTNetworkUsageManagerInterface)interface;
+
 - (MTSignal *)currentStats;
 
 @end
diff --git a/MTNetworkUsageManager.m b/MTNetworkUsageManager.m
index 7c8294a4e7..de9e4b7cc3 100644
--- a/MTNetworkUsageManager.m
+++ b/MTNetworkUsageManager.m
@@ -1,12 +1,25 @@
 #import "MTNetworkUsageManager.h"
 
-#import "MTNetworkUsageCalculationInfo.h"
-#import "MTTimer.h"
-#import "MTQueue.h"
-#import "MTSignal.h"
 #include <sys/mman.h>
 #import <libkern/OSAtomic.h>
 
+#if defined(MtProtoKitDynamicFramework)
+#   import <MTProtoKitDynamic/MTNetworkUsageCalculationInfo.h>
+#   import <MTProtoKitDynamic/MTSignal.h>
+#   import <MTProtoKitDynamic/MTTimer.h>
+#   import <MTProtoKitDynamic/MTQueue.h>
+#elif defined(MtProtoKitMacFramework)
+#   import <MTProtoKitMac/MTNetworkUsageCalculationInfo.h>
+#   import <MTProtoKitMac/MTSignal.h>
+#   import <MTProtoKitMac/MTTimer.h>
+#   import <MTProtoKitMac/MTQueue.h>
+#else
+#   import <MTProtoKit/MTNetworkUsageCalculationInfo.h>
+#   import <MTProtoKit/MTSignal.h>
+#   import <MTProtoKit/MTTimer.h>
+#   import <MTProtoKit/MTQueue.h>
+#endif
+
 @implementation MTNetworkUsageManagerStats
 
 - (instancetype)initWithWWAN:(MTNetworkUsageManagerInterfaceStats)wwan other:(MTNetworkUsageManagerInterfaceStats)other {
@@ -95,6 +108,24 @@ static int interfaceOffset(MTNetworkUsageManagerInterface interface) {
     }];
 }
 
+- (void)resetIncomingBytes:(MTNetworkUsageManagerInterface)interface {
+    [_queue dispatchOnQueue:^{
+        if (_map) {
+            int64_t *ptr = (int64_t *)(_map + 0 * 16 + interfaceOffset(interface));
+            *ptr = 0;
+        }
+    }];
+}
+
+- (void)resetOutgoingBytes:(MTNetworkUsageManagerInterface)interface {
+    [_queue dispatchOnQueue:^{
+        if (_map) {
+            int64_t *ptr = (int64_t *)(_map + 1 * 16 + interfaceOffset(interface));
+            *ptr = 0;
+        }
+    }];
+}
+
 - (MTSignal *)currentStats {
     return [[MTSignal alloc] initWithGenerator:^id<MTDisposable>(MTSubscriber *subscriber) {
         [_queue dispatchOnQueue:^{