(wip) fix with_puppet_running_on