unit_test update for compilation