Massa surprised to see Heidfeld in test driver role