2 files changed, 11 insertions, 0 deletions

diff --git a/spec/inputs/assign.yue b/spec/inputs/assign.yue
index ad1eda3..74583eb 100644
--- a/spec/inputs/assign.yue
+++ b/spec/inputs/assign.yue
@@ -127,4 +127,9 @@ do
         a |= 7
         a ^= 8
 
+do
+        type or= (v) ->
+                type = type v
+                type
+
 nil
diff --git a/spec/outputs/assign.lua b/spec/outputs/assign.lua
index 89c5f8a..c96bf00 100644
--- a/spec/outputs/assign.lua
+++ b/spec/outputs/assign.lua
@@ -204,5 +204,11 @@ return _(function()
                 a = a | 7
                 a = a ^ 8
         end
+        do
+                local type = type or function(v)
+                        local type = type(v)
+                        return type
+                end
+        end
         return nil
 end)