From bb25aeeb1ead8eb59b8d825fca7aa1a907038b2a Mon Sep 17 00:00:00 2001
From: Philippe Kirschen <philippe@hyperloop-one.com>
Date: Tue, 6 Jul 2021 09:52:30 -0700
Subject: [PATCH] another plot_fit fix

---
 gpfit/plot_fit.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/gpfit/plot_fit.py b/gpfit/plot_fit.py
index a4887fa..1dba6d7 100644
--- a/gpfit/plot_fit.py
+++ b/gpfit/plot_fit.py
@@ -24,11 +24,11 @@ def plot_fit_1d(udata, wdata, K=1, fitclass='MA', plotspace='log'):
 
 
     if fitclass == 'SMA':
-        wexps, = cstrt.left.exps
-        alpha, = list(wexps.values())
-        uvarkey, = cstrt.right.varkeys
-        A = [d[uvarkey]/alpha for d in cstrt.right.exps]
-        B = np.log(cstrt.right.cs) / alpha
+        wexps, = cstrt[0].left.exps
+        alpha = list(wexps.values())[0]
+        uvarkey, = cstrt[0].right.varkeys
+        A = [d[uvarkey]/alpha for d in cstrt[0].right.exps]
+        B = np.log(cstrt[0].right.cs) / alpha
 
         ww = 0
         for k in range(K):