Fix tests for volume migration tasks