Fixes state update for migration task

Fixes EUCA-6159