Fix focus error after context view is closed