Migration task fix