modified unit test to be more specific