Update test to be more specific