El delantero internacional español terminó la temporada 2016/17 con un título de Premier bajo el brazo, siendo además, el máximo goleador del Chelsea. En sus vacaciones recibió un mensaje de su entrenador, Antonio Conte de que no contaba con él para la próxima temporada y Costa comunicó al club que no volvería a entrenar a las órdenes del técnico italiano, y que si no contaban con él que sólo quería ser traspasado al Atlético de Madrid.