*

1 files changed, 7 insertions, 1 deletions

diff --git a/lua/chupacabra/scratch.lua b/lua/chupacabra/scratch.lua
index 4172950..6025888 100644
--- a/lua/chupacabra/scratch.lua
+++ b/lua/chupacabra/scratch.lua
@@ -22,7 +22,11 @@ end
 local function print_stack()
     for i, entry in ipairs(stack) do
         print(i .. "  input: " .. entry.input)
-        print("  output: " .. table_to_string(entry.output))
+        if type(entry.output) == "table" then
+            print("  output: " .. table_to_string(entry.output))
+        else
+            print("  output: " .. tostring(entry.output))
+        end
     end
 end
 
@@ -31,4 +35,6 @@ scratch("[2 4 6] 2 @..")
 scratch("3 : + [1 1 1] [2 3 4] @+ @..")
 scratch("[2 4 6] 2 @..")
 scratch("1 2 3 4 5 6 7 @.. [1 2 3 0 5 6 7] @=")
+scratch("2 3 +")
+scratch("3 : + 6 =")
 print_stack()