Updated test assert calls.