add context to some tests