diff --git a/composer/utils/warnings.py b/composer/utils/warnings.py
index 72c3be3e67..5a8c31c407 100644
--- a/composer/utils/warnings.py
+++ b/composer/utils/warnings.py
@@ -14,6 +14,7 @@ class VersionedDeprecationWarning(DeprecationWarning):
         remove_version (str): The version in which the feature will be removed.
 
     Example:
+        >>> import warnings
         >>> def deprecated_function():
         ...     warnings.warn(
         ...         VersionedDeprecationWarning(