Fix small template issue