Diego Costa đã hoàn tất những thủ tục cần thiết để trở trở lại Atletico Madrid thi đấu. Tuy nhiên do đội bóng Tây Ban Nha vẫn đang trong thời gian chịu án cấm chuyển nhượng của UEFA, nên phải đến 01/2018 thì Costa mới có thể ra sân thi đấu.
Trong khoảng thời gian này, nhằm giúp Costa giữ duy trì được thể trạng, cảm giác bóng,... Atletico cũng đã cho anh tới sân tập luyện cùng các đồng đội. Thậm chí tiền đạo 29 tuổi này còn được tập riêng cùng HLV thể lực Oscar Ortega.
Chưa dừng lại ở đó, Fernando Torres cũng đã chủ động đề nghị giúp đỡ Costa trong khoảng thời gian này. Cụ thể El Nino đã cho phép Costa tới sử dụng và tập luyện tại phòng tập riêng của mình là NineFitness.
Tại NineFitness, Torres đã giới thiệu Costa với chuyên gia Jesus Garcia Pola, người sẽ cố vấn cho Costa những bài tập cụ thể để giúp tiền đạo người Tây Ban Nha nâng cao sự linh hoạt trong cách di chuyển.