My car got some water in the right headlight during the winter so now I am shopping for a new headlight assembly, The part number to my current one is 8K0 941 030 AS & that one is a bit pricey, I see some of the same design for $150-$250 with a different part number like this one: 8K0941030AL

So do I need to stick with the part number on my car currently or is any headlight assembly with the same curved LED design okay to plug and play? I have a 2010 Audi A4 Premium Plus, thanks!