Fix test regression