diff --git a/src/main/java/com/google/genai/ChatBase.java b/src/main/java/com/google/genai/ChatBase.java
index 5f548c62749..79e58898d88 100644
--- a/src/main/java/com/google/genai/ChatBase.java
+++ b/src/main/java/com/google/genai/ChatBase.java
@@ -37,8 +37,8 @@ class ChatBase {
   private static final Logger logger = Logger.getLogger(ChatBase.class.getName());
 
   ChatBase(List<Content> comprehensiveHistory, List<Content> curatedHistory) {
-    this.comprehensiveHistory = new ArrayList<>();
-    this.curatedHistory = new ArrayList<>();
+    this.comprehensiveHistory = comprehensiveHistory;
+    this.curatedHistory = curatedHistory;
   }
 
   /**