diff --git a/flow360/__init__.py b/flow360/__init__.py
index 3ade4723e..9d6d2dd40 100644
--- a/flow360/__init__.py
+++ b/flow360/__init__.py
@@ -89,6 +89,7 @@
     FromUserDefinedDynamics,
     HeatEquationInitialCondition,
     NavierStokesInitialCondition,
+    NavierStokesModifiedRestartSolution,
     PorousMedium,
     Rotation,
     Solid,
@@ -239,6 +240,7 @@
     "UserDefinedDynamic",
     "Translational",
     "NavierStokesInitialCondition",
+    "NavierStokesModifiedRestartSolution",
     "FromUserDefinedDynamics",
     "HeatEquationInitialCondition",
     "Temperature",
diff --git a/flow360/plugins/report/__init__.py b/flow360/plugins/report/__init__.py
index c30755da1..feaceb736 100644
--- a/flow360/plugins/report/__init__.py
+++ b/flow360/plugins/report/__init__.py
@@ -1,7 +1,6 @@
 """Utilities for creating reports"""
 
 from flow360.plugins.report.report import ReportTemplate
-from flow360.plugins.report.report_context import ReportContext
 from flow360.plugins.report.report_items import (
     Chart2D,
     Chart3D,
@@ -58,7 +57,6 @@
     "RearCamera",
     "RearLeftTopCamera",
     "RearRightBottomCamera",
-    "ReportContext",
     "ReportTemplate",
     "Settings",
     "SubsetLimit",