minor import bug fix