(sort of) fix failing test