Added missng class import.