Adjust expectation to match exising code
Adjust test expectation of underlying API