Fixing focus bug.