Fixed broken test by last commit