updated the unit_tests