Fix test data setup