There are a number of problems with this sequence as it stands and in both of your edits.
Some people argue that Force Lightning should be kept for Palpatine alone as a sign of his unique mastery of the dark side.
That aside if he can use that power against Yoda, why doesn't he use it against Obi-Wan or Anakin in ROTS (or indeed earlier here).
In your removing of the Yoda it begs the question why leave Obi-Wan and Anakin alive?
One solution is remove the force lightning and have Dooku throw Anakin against the wall with a Force push (briefly knocking him out as he does with Obi-Wan in ROTS).
Then keep Yoda in but remove the Force lightning elements so he throws objects at Yoda (Vader and Maul can do this so it makes sense that Dooku can) and finishes with trying to drop the column on Obi-Wan and Anakin and Dooku uses the distraction to escape.
That way Dooku's powers are consistent in both films (and it also leaves Force Lightning as a special power for Palpatine).