diff --git a/examples/demo_cffi.py b/examples/demo_cffi.py
index 4347ec9..cb462e6 100644
--- a/examples/demo_cffi.py
+++ b/examples/demo_cffi.py
@@ -2,7 +2,7 @@
 import sys
 sys.path.append(os.path.abspath('.'))
 
-from llama.llama_cli_cffi import llama_generate, LlamaOptions
+from llama import llama_generate, LlamaOptions
 
 
 options = LlamaOptions(
diff --git a/examples/demo_ctypes.py b/examples/demo_ctypes.py
index 781126b..8cbfdfc 100644
--- a/examples/demo_ctypes.py
+++ b/examples/demo_ctypes.py
@@ -2,7 +2,7 @@
 import sys
 sys.path.append(os.path.abspath('.'))
 
-from llama.llama_cli_ctypes import llama_generate, LlamaOptions
+from llama.ctypes import llama_generate, LlamaOptions
 
 
 options = LlamaOptions(
diff --git a/llama/__init__.py b/llama/__init__.py
index 83ba590..6d73d50 100644
--- a/llama/__init__.py
+++ b/llama/__init__.py
@@ -1,2 +1,2 @@
-# from .llama_cpp_ctypes import *
-# from .llama_cpp_ctypes_options import *
\ No newline at end of file
+from .llama_cli_cffi import *
+from .llama_cli_options import *
diff --git a/llama/ctypes.py b/llama/ctypes.py
new file mode 100644
index 0000000..8fc25e3
--- /dev/null
+++ b/llama/ctypes.py
@@ -0,0 +1,2 @@
+from .llama_cli_ctypes import *
+from .llama_cli_options import *
\ No newline at end of file