Fix unitary tests for update