Move context to tests