Fixes unittest regression.