Adding a div to fix the breakage