diff --git a/locust/__init__.py b/locust/__init__.py
index a78ea9913d..199432ce49 100644
--- a/locust/__init__.py
+++ b/locust/__init__.py
@@ -26,6 +26,7 @@
     "constant",
     "constant_pacing",
     "events",
+    "LoadTestShape",
 )
 
 # Used for raising a DeprecationWarning if old Locust/HttpLocust is used