"Want to ensure there is no discomfort when I return" - Mohammed Shami aims to make comeback only after regaining full fitness