fixed one failing test