Updating unit-test