Fix missing div at end