immproved unit test