Listen to keypress in ve.ce.surface.$ rather than window

Bug: 50538
Change-Id: Ifddf1bae1ee799ec3467a81c597702fc3f8be089