fix test after function rename