fixed test breackage