fix test regression