Fixing waveform display in assignment