fixes failing test