Rohit said the team management wants Shami to regain full fitness before returning to top-flight cricket. Rohit said the National Cricket Academy staff will continue to assess Shami's fitness and he might play a couple of internal matches in that process.