Pass branch to checkout_module function.

Fix the puppet::install task so it actually passes the branch arg.