Fix compilation error in unit test due to wrong import.