From 19d024b818722187a978f0f463c56d7730c10f42 Mon Sep 17 00:00:00 2001
From: Stuart Mumford <stuart@cadair.com>
Date: Wed, 20 Nov 2024 16:11:02 +0000
Subject: [PATCH] Fix a naxes bug

---
 gwcs/api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gwcs/api.py b/gwcs/api.py
index bcff2c1b..fed751ff 100644
--- a/gwcs/api.py
+++ b/gwcs/api.py
@@ -65,7 +65,7 @@ def world_axis_units(self):
 
     def _remove_quantity_output(self, result, frame):
         if self.forward_transform.uses_quantity:
-            if self.output_frame.naxes == 1:
+            if frame.naxes == 1:
                 result = [result]
 
             result = tuple(r.to_value(unit) if isinstance(r, u.Quantity) else r