update test to refine setup