Update unit-tests