solve the unit test problem