fixing failing test