From 70230b5b833a9a1086484f1c025a4bca621b3fc9 Mon Sep 17 00:00:00 2001
From: Luke Evers <luke@lukevers.com>
Date: Wed, 20 Dec 2017 15:23:30 -0500
Subject: [PATCH] Limit surrogate keys to template keys if 404 page

---
 src/classes/surrogate-key-collection.php | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/classes/surrogate-key-collection.php b/src/classes/surrogate-key-collection.php
index 4dfed3d..57a3d2a 100644
--- a/src/classes/surrogate-key-collection.php
+++ b/src/classes/surrogate-key-collection.php
@@ -52,6 +52,10 @@ public function __construct($wp_query)
             $term_keys
         );
 
+        if (in_array('tm-404', $template_key)) {
+            $keys = $template_key;
+        }
+
         $keys = array_unique($keys);
         $keys = array_filter($keys);