Fix overly specific test