updated example into unit test