Small import change