Fix changed import.