diff --git a/src/sagemaker/rl/__init__.py b/src/sagemaker/rl/__init__.py
index 62a0c1e934..562cf863ed 100644
--- a/src/sagemaker/rl/__init__.py
+++ b/src/sagemaker/rl/__init__.py
@@ -12,7 +12,8 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+__all__ = [
+    'RLEstimator', 'RLToolkit', 'RLFramework', 'TOOLKIT_FRAMEWORK_VERSION_MAP']
+
 from sagemaker.rl.estimator import (RLEstimator, RLToolkit, RLFramework,
                                     TOOLKIT_FRAMEWORK_VERSION_MAP)
-
-__all__ = [RLEstimator, RLToolkit, RLFramework, TOOLKIT_FRAMEWORK_VERSION_MAP]