Fix #318 - Missing end div causing layout issues

Fixes #318