Checkmate button fix