Updated test assert statements.