Make UT more easy to maintain