My Passport - WXD1A17JJJVN

I am looking to replace the head stack assembly on my “WD My Passport 2TB - WD20NMVW”. The broken drive was manufactured in Jan 2017 and it has 4 discs/platters so I was hoping someone could tell me what WD hard drive has the same head stack assembly that I can swap out and replace the defective head?