Update puppet tests