diff --git a/cuda_malloc.py b/cuda_malloc.py
index fd318e30b..1204df90c 100644
--- a/cuda_malloc.py
+++ b/cuda_malloc.py
@@ -76,7 +76,7 @@ try:
             module = importlib.util.module_from_spec(spec)
             spec.loader.exec_module(module)
             version = module.__version__
-            cuda_available = module.cuda != None
+            cuda_available = module.cuda is not None
 except:
     pass