Fixes the focus problem.