Fixing missing div tag