Fixed a bug in the 3D wave field