fix test after changes in return value