Fixing test after latest changes