Fixed reverting focus (wasn't right last time)