Fix test to be more specific