Update UI correctly on CMD+K