small fix for output