diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py
index 22fb841fb..02ca804ae 100644
--- a/bitsandbytes/__init__.py
+++ b/bitsandbytes/__init__.py
@@ -4,7 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 
 from .nn import modules
-from cextension import COMPILED_WITH_CUDA
+from .cextension import COMPILED_WITH_CUDA
 
 if COMPILED_WITH_CUDA:
     from .optim import adam
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
index 68b1d7883..ab4e565a9 100644
--- a/bitsandbytes/functional.py
+++ b/bitsandbytes/functional.py
@@ -9,7 +9,7 @@
 import torch
 from torch import Tensor
 
-from cextension import lib, COMPILED_WITH_CUDA
+from .cextension import lib, COMPILED_WITH_CUDA
 
 name2qmap = {}
 
diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py
index e833ecc23..42b5bc0b9 100644
--- a/bitsandbytes/optim/__init__.py
+++ b/bitsandbytes/optim/__init__.py
@@ -13,4 +13,5 @@
     from .lamb import LAMB, LAMB8bit, LAMB32bit
     from .rmsprop import RMSprop, RMSprop8bit, RMSprop32bit
     from .adagrad import Adagrad, Adagrad8bit, Adagrad32bit
-    from .optimizer import GlobalOptimManager
+
+from .optimizer import GlobalOptimManager