Changed trait to context.