Any DI should work. You might have to swap electronics though. I am no DI expert. Also the DI is the most finicky engine they made so make sure you find what caused the failure and I wouldn't buy a used engine.
If you are looking to get back on the water fast with a warranty SBT is about your only option.