Add more UT cases