fix last test regression