fix break in test