Add do_puppet_run command for both versions