Fix to work with puppet 3.4