Remove unused import.
Updated to reflect upstream API changes.