Fix test data working problem