Fix 2d surface in old plotting