Get focus working again