diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index b5e3e64fd..eb6c0a00f 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -80,8 +80,8 @@ jobs:
           poetry run ruff check . --fix --exit-zero
           poetry run ruff check .
 
-      - name: Typecheck with pyright
-        uses: jakebailey/pyright-action@v2
+      - name: Run pyright
+        run: poetry run pyright
 
       - name: Thorough check with pylint
         run: poetry run pylint awpy
diff --git a/awpy/__init__.py b/awpy/__init__.py
index 9e41214b7..5e4e84b03 100644
--- a/awpy/__init__.py
+++ b/awpy/__init__.py
@@ -1,4 +1,5 @@
 """Provides data parsing, analytics and visualization capabilities for CSGO data."""
-
+from awpy.demo import Demo
 
 __version__ = "2.0.0-alpha"
+__all__ = ["Demo"]