All they need to do is replace or repair the part that broke under warranty and it sounds like they are doing that. Lemon laws are for chronic problems that cannot seem to be fixed. An engine sounds like a big deal, but it's just a part that broke and needs to be replaced.
I wouldn't sweat it.
I wouldn't sweat it.