diff --git a/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java b/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java
index 7d6eb6131..fc7d70445 100644
--- a/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java
+++ b/src/main/java/ru/vk/itmo/shemetovalexey/InMemoryDao.java
@@ -88,7 +88,6 @@ public void upsert(Entry<MemorySegment> entry) {
             long valueSize = entry.value() == null ? 0 : entry.value().byteSize();
             if (size.addAndGet(keySize + valueSize) >= maxSize) {
                 flush();
-                size.set(0);
             }
         } catch (IOException e) {
             throw new UncheckedIOException(e);
@@ -177,6 +176,7 @@ public void flush() throws IOException {
             } finally {
                 upsertLock.writeLock().unlock();
             }
+            size.set(0);
         });
     }