Update handler: fix can't check if done