fix regression in test