Adapt to new jquery-ui structure