SSD replacement on Surface Pro

Hi guys.

Hope someone could give me a bit of advice regarding Surface Pro SSD replacement.

I was given a 1st Gen Surface Pro 256GB tablet with a dead SSD for free. The unit is in excellent condition and boxed, so I was quite keen to repair it. Using the iFixit guide I managed to remove the screen/digitizer without damaging the assembly or the ribbons.

Now my question is this - can I use any make msata SSD to replace the dead one? Are there any makes I should stay away from for reasons such as bios detection issues, booting problems, etc...

Thanks in advance.