Yes, that makes sense. He probably transfered a little of his power, but had to continue the effort after his temporary death.

I'm glad you liked it, and yes I wish this was the canon version.