Fixing focus behavior.