Fix test to correspond to actual behavior