Focuses correctly after route change