The fast bowler was a late replacement for Jasprit Bumrah, and had to work on his fitness after having Covid