fix output function