correcting the output