diff --git a/rbproto/__init__.py b/rbproto/__init__.py
index 5de8af2..397f57e 100644
--- a/rbproto/__init__.py
+++ b/rbproto/__init__.py
@@ -1 +1 @@
-from .python import *
+from .python.shot_pb2 import *  # noqa: F401