fixing failing unittest