Fix modal focus fun