#7 fixing failing test