scratch: fixed pad multiple screen support

Unmanage handler was deleting the wrong client for the scratch
table. Causing lua tracebacks and breaking awesome on next scratch.pad
call. Now proper scratched clients are zeroed in multi display setups.

1 files changed, 3 insertions, 2 deletions

diff --git a/scratch/pad.lua b/scratch/pad.lua
index cdf87e0..920d1d6 100644
--- a/scratch/pad.lua
+++ b/scratch/pad.lua
@@ -24,6 +24,7 @@
 ---------------------------------------------------------------
 
 -- Grab environment
+local pairs = pairs
 local awful = require("awful")
 local capi = {
     mouse = mouse,
@@ -79,8 +80,8 @@ function set(c, width, height, sticky, screen)
     if not scratchpad.pad then scratchpad.pad = {}
         -- add unmanage signal for scratchpad clients
         capi.client.add_signal("unmanage", function (c)
-            if scratchpad.pad[screen] == c then
-                scratchpad.pad[screen] = nil
+            for scr, cl in pairs(scratchpad.pad) do
+                if cl == c then scratchpad.pad[scr] = nil end
             end
         end)
     end