fix output issue