Fix #318 - Missing end div causing layout issues