Minor import change