diff --git a/third_party/proton/proton/viewer.py b/third_party/proton/proton/viewer.py
index 84591f95e461..f77a65007fc7 100644
--- a/third_party/proton/proton/viewer.py
+++ b/third_party/proton/proton/viewer.py
@@ -4,10 +4,10 @@
 import pandas as pd
 try:
     import hatchet as ht
+    from hatchet.query import NegationQuery
 except ImportError:
     raise ImportError("Failed to import hatchet. `pip install llnl-hatchet` to get the correct version.")
 import numpy as np
-from hatchet.query import NegationQuery
 from triton.profiler.hook import COMPUTE_METADATA_SCOPE_NAME, TritonHook