Batting in the second innings was what let India down in the first Test match against Sri Lanka, and now Virat Kohli's men will look to put that right when Day 4 of the second Test match comes rolling at the P Sara Oval.
Sitting pretty at 70/1 with a lead of 157 runs, India are in a similar position of strength they were in in the first Test, before they let it all slip away.
Day 4 will be about batting well and finding that right balance between aggression and don't-be-daft, and if the India batsmen can do that, they will setup a target that the bowlers will relish playing with going into the final day of this Test match in Colombo.
Ajinkya Rahane and Murali Vijay are batting at the crease and India will look to get to a score that they think will be enough on this still-pretty-decent-for-batting wicket.
"[We would like] another 250-300 runs and 100 overs to bowl at the Sri Lankans," said Amit Mishra, who was the highest wicket-taker in the first innings, picking up four.
"Even now, the wicket is good for batting because it is still a bit slow. [We will look to] bat for as long as possible so that on the final day, we can see what happens.
"No doubt it is helping the spinners a bit but because there is not much pace, it is still decent for batting. It is possible that on the final day, it might be difficult to bat on."
Sri Lanka will know a couple of early wickets and they can cause a bit of panic in the India camp. The last time India were put under pressure, they did not handle it too well, and the key will be for the batsmen who are settled to stay at the crease for a long time and build that all-crucial partnership.
No doubt, the spinners Rangana Herath and Tharindu Kaushal will have a big role to play for Sri Lanka on Day 4 if the hosts are to restrict India to a small total, and the first session could prove to be extremely crucial.
Where to Watch Live
Get the Day 5 Live Blog HERE
Get Day 4 As It Happened HERE
India vs Sri Lanka is set to begin at 10 am local time (10 am IST, 5.30 am BST, 12.30 am ET). Live coverage in India will be on Sony Six (English) and Sony Kix (Hindi). Viewers in Sri Lanka and the rest of the subcontinent can also catch the match live on Sony Six.
Audience watching the match by staying up late in the US can switch to Willow TV, while the live stream option is on Willow TV Online.
SuperSport and SuperSport Online will provide the live and live stream options in South Africa and Sub Saharan Africa.