240 Searay Sundancer with 5.7 electronic fuel injection

Just had engine replaced and used the original fuel injection. The engine runs fine after starting and taking it on a run. It's when coming off a cruise it will shut off and need to be re started.