fix output for modified function