Move imports into tests
so travis won't complain