add back collation (still the way it was working before)