That's basically what Musk said, yes: that they need to work on both engine performance and performance consistency. He said they'd be cutting off the main engines one second sooner to provide more margin for boost-back.

Amazing that a one-second difference can matter, but those things go through tons of fuel every one of those seconds!