Fix array shape error in incompletely initialized wavepackets