minor import adjustment