From ea2386a84bd774084b1c477d1f244312a9dac9e1 Mon Sep 17 00:00:00 2001
From: Yoel Cortes-Pena <yoelcortes@gmail.com>
Date: Fri, 24 May 2024 11:05:35 -0500
Subject: [PATCH] fix test

---
 flexsolve/profiler.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/flexsolve/profiler.py b/flexsolve/profiler.py
index 8323a59..c072889 100644
--- a/flexsolve/profiler.py
+++ b/flexsolve/profiler.py
@@ -72,7 +72,10 @@ def __call__(self, x, *args):
         x = copy(x)
         self.xs.append(x)
         y = self.f(x, *args)
-        self.ys.append(y)
+        if isinstance(y, tuple): 
+            self.ys.append(y[0])
+        else:
+            self.ys.append(y)
         return y
     
     def archive(self, name):