diff --git a/Partner-Core/src/main/java/work/slhaf/partner/core/memory/MemoryCore.java b/Partner-Core/src/main/java/work/slhaf/partner/core/memory/MemoryCore.java
index 28c4dbc2..ddb71263 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/core/memory/MemoryCore.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/core/memory/MemoryCore.java
@@ -52,6 +52,7 @@ public class MemoryCore implements StateSerializable {
             conversationMessages.addAll(chatMessages);
 
             unit.getSlices().add(memorySlice);
+            normalizeMemoryUnit(unit);
             return unit;
         } finally {
             memoryLock.unlock();