fix output problem