Update test to avoid failures