Costa's future at Stamford Bridge is in doubt following a bust-up with Antonio Conte and interest from China.
The striker returned to the side yesterday as Chelsea beat Hull 2-0 to extend their lead in the title race.
And international team-mate Pedro says the 28-year-old is happy at Chelsea and is showing the goods on the training ground.
Diego Costa is back in the Chelsea team and back among the goals
Diego Costa rages at Pedro after poor pass against Tottenham
Wed, January 4, 2017
Diego Costa was not impressed with team-mate Pedro during Chelsea's match against Tottenham…
1 of 6
Diego Costa was not impressed with Pedro tonight…
“With the team, he has been training very well," Pedro said.
"He is very happy to be here in the club, this is the most important thing.
Diego Costa now has 15 Premier League goals for the season
“His focus in the team, the game, this season to fight for this title.
"It’s good for the team. This ambition Diego shows is good for the team.”