need to do the saving in AdamContext