Fix focus problem