diff --git a/Github_Commit_Whitespace/Github_Commit_Whitespace.user.js b/Github_Commit_Whitespace/Github_Commit_Whitespace.user.js
index 1b58d58..407bd4d 100644
--- a/Github_Commit_Whitespace/Github_Commit_Whitespace.user.js
+++ b/Github_Commit_Whitespace/Github_Commit_Whitespace.user.js
@@ -49,7 +49,7 @@
             g.appendChild(a)
 
             b.parentNode.insertBefore(g, b)
-        } else if (/\/pull\/\d*\/(files|commits)/.test(location.href) && (e = document.querySelector('#files_bucket .pr-toolbar .diffbar > .float-right'))) {
+        } else if (/\/pull\/\d*\/(files|commits)/.test(location.href) && (e = document.querySelector('#files_bucket .pr-toolbar .diffbar > .pr-review-tools'))) {
 
             var r = e.querySelector('.GithubCommitWhitespaceButton')
             if (r) {