diff --git a/python/examples/cutlass_gemm.py b/python/examples/cutlass_gemm.py
index 1675d0c..154bc16 100644
--- a/python/examples/cutlass_gemm.py
+++ b/python/examples/cutlass_gemm.py
@@ -19,12 +19,11 @@ import sys
 
 import cuda.bindings.driver as driver
 import cuda.core.experimental as core
+import cuda.nvbench as nvbench
 import cupy as cp
 import cutlass
 import numpy as np
 
-import nvbench
-
 
 def as_bindings_Stream(cs: nvbench.CudaStream) -> driver.CUstream:
     return driver.CUstream(cs.addressof())