This is what I upgraded my Omen 16 with
https://www.amazon.com/dp/B094WB68CY?psc=1&ref=ppx\_yo2\_dt\_b\_product\_details