Yes, they have to repair it under warranty. The Magnuson-Moss Warranty Act protects you here (assuming you’re USA). I would definitely swap back your old drive unless they guarantee that you’re getting the same one back. If you send it in they may send you a different one that doesn’t have your drive in it.