diff --git a/mlonmcu/context/__init__.py b/mlonmcu/context/__init__.py
index db85df904..c58af4283 100644
--- a/mlonmcu/context/__init__.py
+++ b/mlonmcu/context/__init__.py
@@ -16,3 +16,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+
+from .context import MlonMcuContext
+
+__all__ = ["MlonMcuContext"]