Fix error in calculation of work group size