Updated test assert