diff --git a/solvebio/query.py b/solvebio/query.py index 7a71c1ab..11f70d3b 100644 --- a/solvebio/query.py +++ b/solvebio/query.py @@ -826,8 +826,7 @@ def migrate(self, target, follow=True, **kwargs): if follow: # Multiple migration objects returned if "data" in migration: - for mig in migration: - mig.follow() + self.activity(follow=True) else: migration.follow() diff --git a/solvebio/resource/dataset.py b/solvebio/resource/dataset.py index 135954e4..451f9034 100644 --- a/solvebio/resource/dataset.py +++ b/solvebio/resource/dataset.py @@ -258,8 +258,7 @@ def migrate(self, target, follow=True, **kwargs): if follow: # Multiple migration objects returned if "data" in migration: - for mig in migration: - mig.follow() + self.activity(follow=True) else: migration.follow()