Added missing class import.