Correct switch statement for dim=3

The old formula contained some 'dim's instead of '2's,
which obviously works for 2D....