diff --git a/tosfs/__init__.py b/tosfs/__init__.py
index 8b29843..6b17404 100644
--- a/tosfs/__init__.py
+++ b/tosfs/__init__.py
@@ -1 +1,3 @@
-from .core import TosFileSystem, TosFile
+from tosfs.core import TosFile, TosFileSystem
+
+__all__ = ["TosFileSystem", "TosFile"]