Trying to fix output divergence.