minor import change