fix issue with the  output