Heikkilä toimi Suomen jääkiekkomaajoukkueen päävalmentajana kauden 1976–1977. Hän johti Suomen peliä MM-kisoissa 1977 ja Kanada-cupissa 1976. Heikkilä palasi maajoukkueeseen vuosiksi 1984–1987, jolloin hän toimi joukkueenjohtajana.