diff --git a/llmfoundry/__init__.py b/llmfoundry/__init__.py
index 6da0cb59c5..b851aaa559 100644
--- a/llmfoundry/__init__.py
+++ b/llmfoundry/__init__.py
@@ -36,7 +36,6 @@
 
 logger.addFilter(new_files_warning_filter)
 
-from llmfoundry._version import __version__
 from llmfoundry import (
     algorithms,
     callbacks,
@@ -51,6 +50,7 @@
     tokenizers,
     utils,
 )
+from llmfoundry._version import __version__
 from llmfoundry.data import StreamingFinetuningDataset, StreamingTextDataset
 from llmfoundry.eval import InContextLearningDataset, InContextLearningMetric
 from llmfoundry.models.hf import ComposerHFCausalLM
@@ -64,6 +64,7 @@
 from llmfoundry.optim import DecoupledLionW
 
 __all__ = [
+    '__version__',
     'StreamingFinetuningDataset',
     'StreamingTextDataset',
     'InContextLearningDataset',