Removed an unnecessary class import.