Changing import to avoid breaking subclassed.