fixing test broken by function change