Set correct category