Fixed import issue for running remote tests.