lib/cli/kit/executor.rb in cli-kit-3.0.0 vs lib/cli/kit/executor.rb in cli-kit-3.0.1

@@ -26,13 +26,17 @@
           end
         end
       end
 
       def twrap(signal, handler)
-        prev_handler = trap(signal, handler)
-        yield
-      ensure
-        trap(signal, prev_handler)
+        return yield unless Signal.list.key?(signal)
+
+        begin
+          prev_handler = trap(signal, handler)
+          yield
+        ensure
+          trap(signal, prev_handler)
+        end
       end
 
       def quit_handler(_sig)
         z = caller
         CLI::UI.raw do