Fixed broken test by lasta commit