Set corect category