Fix assertTemplateName() checking only the first leaf in the view model tree.

Fix assertTemplateName() checking only the first leaf in the view model tree. #399