update for puppet support