diff --git a/python/cutlass_library/heuristics.py b/python/cutlass_library/heuristics.py
index dc69e103..83421a06 100644
--- a/python/cutlass_library/heuristics.py
+++ b/python/cutlass_library/heuristics.py
@@ -37,7 +37,8 @@ import json
 import csv
 
 try:
-  if CUTLASS_IGNORE_PACKAGE:
+  import builtins
+  if hasattr(builtins, "CUTLASS_IGNORE_PACKAGE") and CUTLASS_IGNORE_PACKAGE == True:
     raise ImportError("Disabling attempt to import cutlass_library")
   from cutlass_library.library import *
   from cutlass_library.generator import *
diff --git a/python/cutlass_library/heuristics_provider.py b/python/cutlass_library/heuristics_provider.py
index 9baff5c0..b3f6e5c5 100644
--- a/python/cutlass_library/heuristics_provider.py
+++ b/python/cutlass_library/heuristics_provider.py
@@ -41,7 +41,13 @@ import logging
 import ctypes
 import functools
 
-from library import DataType, LayoutType
+try:
+  import builtins
+  if hasattr(builtins, "CUTLASS_IGNORE_PACKAGE") and CUTLASS_IGNORE_PACKAGE == True:
+    raise ImportError("Disabling attempt to import cutlass_library")
+  from cutlass_library.library import DataType, LayoutType
+except ImportError:
+  from library import DataType, LayoutType
 
 class MatmulHeuristics: