Correcting the height of the boxes