fix output plot