fix test function return value