correcting unit-tests