Fix handling of sub-slabs in OpenMP code