Fix focus/blur behavior