But Shastri has warned Australia and said Kohli is back in the country where he loves to bat and score runs.