fixing test regression