Fix 3D plot