fix output bugs