add a small test in unit test