[Migration] remove set_device()