Fix task generation in the arithmetic task