Adding missing import to driver template.