changed test to correspond to new data