Fix a bit of a div mess