Add missing @CheckForNull on templateKey