Could be injector seals/washers.
Could be fuel filter or housing.
Or fuel pressure regulator.
If it was running fine before pump died and now it doesn't then it may still be pump related, especially if it is a cheap one.
My son had an issue where it didn't start well and ran rough, we changed fuel filter housing and it was fine afterwards.
Fitted a new pump when it cut out last week similar to you, good as new after running a purge cycle.