Last test regression fix