Add template set check