Update test setup