[PATCH] Ensure WKD key lookup returns the correct key