Fixed after test regression