Fix import extendWorkload for update