fixed test broken with last commit