The K40 relay module can also cause the trouble. I'd try the CPS first, even though the K40 is easier to change. Unless you know that the fuel pump is shutting down before the engine, in which case I'd probably do the K40 first.