Updated for puppet module v4.0.0