Editor Add-link button in WP 3.1

I find that in WP 3.1 the “add-link” button in rich editor in my plugin doesn’t work, due to new WP dialog (with internal linking).
If someone knows how to fix it please share the solution.