You can do what I did and install a Chris King 😂
You would lose the angle adjustment though if you care about that.
With that being said, the stock headset should be just fine as well. Def check what @ebikerider said first, especially since you could check, fix if needed, and retorque it...