Men om det finns skäl till det , ska försäkringskassan ta över ansvaret för rehabiliteringsutredningen .