Small fix with focus