From 71011c2a6f2829280bc6aebbe0839e4156c64edf Mon Sep 17 00:00:00 2001
From: Yingjie Wang <phywyj@gmail.com>
Date: Thu, 6 Nov 2025 16:09:25 -0500
Subject: [PATCH] fix: update sample->capacity

---
 src/init.c | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/init.c b/src/init.c
index e4a3c64..9cabd01 100644
--- a/src/init.c
+++ b/src/init.c
@@ -102,8 +102,10 @@ double chi(double b){
 
 void sample_push(SampleData *sample, double x, double y){
   if (sample->size + 1 >= sample->capacity) {
-    sample->x = realloc(sample->x, sizeof(double) * (sample->capacity)*2);
-    sample->y = realloc(sample->y, sizeof(double) * (sample->capacity)*2);
+    int newcap = sample->capacity*2;
+    sample->x = realloc(sample->x, sizeof(double) * newcap);
+    sample->y = realloc(sample->y, sizeof(double) * newcap);
+    sample->capacity = newcap;
   }
   sample->x[sample->size] = x;
   sample->y[sample->size] = y;