MAINT: bug fix, updated output