diff --git a/__init__.py b/__init__.py
index aed3417..a4c1851 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2,8 +2,10 @@
 
 import struct
 
+from binaryninja.enums import Endianness
+
 from binaryninja import (
-    Architecture, RegisterInfo, InstructionInfo,
+    Architecture, BinaryViewType, RegisterInfo, InstructionInfo,
 
     InstructionTextToken, InstructionTextTokenType,