Set correct expected task type count