nbd: move the task_recv check into nbd_size_update