Issue with locate_dofs_geometrical using V.sub(0).collapse

Collapse also returns the map from the sub space to the parent space, so you need to call V.sub(0).collapse()[0]