fix after minor api change to atlas.