Add container element to fix positioning