Fix omni-search box's behaviour and styling