Focus on button after error