Fixing puppet tests